Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -130,7 +130,7 @@ def predict(image_input, question):
|
|
| 130 |
generated_tokens = model.phi3.generate(
|
| 131 |
input_ids=extended_input_ids,
|
| 132 |
attention_mask=extended_attention_mask,
|
| 133 |
-
max_length=
|
| 134 |
pad_token_id=text_tokenizer.eos_token_id
|
| 135 |
)
|
| 136 |
|
|
@@ -139,9 +139,6 @@ def predict(image_input, question):
|
|
| 139 |
|
| 140 |
return answer
|
| 141 |
|
| 142 |
-
except Exception as e:
|
| 143 |
-
return f"An error occurred: {str(e)}"
|
| 144 |
-
|
| 145 |
except Exception as e:
|
| 146 |
#return f"An error occurred: {str(e)}"
|
| 147 |
return f"An error occurred: {traceback.format_exc()}"
|
|
|
|
| 130 |
generated_tokens = model.phi3.generate(
|
| 131 |
input_ids=extended_input_ids,
|
| 132 |
attention_mask=extended_attention_mask,
|
| 133 |
+
max_length=128,
|
| 134 |
pad_token_id=text_tokenizer.eos_token_id
|
| 135 |
)
|
| 136 |
|
|
|
|
| 139 |
|
| 140 |
return answer
|
| 141 |
|
|
|
|
|
|
|
|
|
|
| 142 |
except Exception as e:
|
| 143 |
#return f"An error occurred: {str(e)}"
|
| 144 |
return f"An error occurred: {traceback.format_exc()}"
|