Spaces:
Running
Running
Update app.py
Browse files
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 |
|