ikraamkb commited on
Commit
cc49774
·
verified ·
1 Parent(s): 6852c86

Update appImage.py

Browse files
Files changed (1) hide show
  1. 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=50)
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)