Monimoy commited on
Commit
67d6f70
·
verified ·
1 Parent(s): 9023859

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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=200,
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()}"