johann22 commited on
Commit
0246362
·
1 Parent(s): adcc88d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -33,7 +33,9 @@ def predict(image):
33
  #inputs = processor(prompts[0], return_tensors="pt").to(device)
34
  print (inputs)
35
 
36
- generated_ids = client.image_to_text(**inputs, max_length=max_length)
 
 
37
  print (generated_ids)
38
 
39
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
33
  #inputs = processor(prompts[0], return_tensors="pt").to(device)
34
  print (inputs)
35
 
36
+ #generated_ids = client.image_to_text(**inputs, max_length=max_length)
37
+ generated_ids = client.image_to_text(inputs, max_length=max_length)
38
+
39
  print (generated_ids)
40
 
41
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]