Spaces:
Running
Running
Update appImage.py
Browse files- appImage.py +1 -1
appImage.py
CHANGED
@@ -67,7 +67,7 @@ async def caption_image(file):
|
|
67 |
if USE_GIT:
|
68 |
image = Image.open(image_path).convert('RGB')
|
69 |
pixel_values = processor(images=image, return_tensors="pt").pixel_values
|
70 |
-
generated_ids = model.generate(pixel_values, max_length=
|
71 |
caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
72 |
else:
|
73 |
captions = captioner(image_path)
|
|
|
67 |
if USE_GIT:
|
68 |
image = Image.open(image_path).convert('RGB')
|
69 |
pixel_values = processor(images=image, return_tensors="pt").pixel_values
|
70 |
+
generated_ids = model.generate(pixel_values, max_length=500)
|
71 |
caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
72 |
else:
|
73 |
captions = captioner(image_path)
|