sanikamal commited on
Commit
2ef1f88
1 Parent(s): b3847c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ def get_format_response(image,question,selected_model):
21
  answer = model.config.id2label[idx]
22
  return answer
23
  else:
 
24
  answer = processor.decode(out[0], skip_special_tokens=True)
25
  return answer
26
 
 
21
  answer = model.config.id2label[idx]
22
  return answer
23
  else:
24
+ outputs = model.generate(**encoding)
25
  answer = processor.decode(out[0], skip_special_tokens=True)
26
  return answer
27