Chintan-Shah commited on
Commit
b3fe912
·
verified ·
1 Parent(s): caa75e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -244,7 +244,7 @@ def getInputs(image_path, question, answer=""):
244
 
245
  return start_input_ids, end_input_ids, image_features, attention_mask
246
 
247
- model_location = "./MM_FT_C1"
248
  # print("Model location:", model_location)
249
 
250
  model = MultimodalPhiModel.from_pretrained(model_location).to(device)
 
244
 
245
  return start_input_ids, end_input_ids, image_features, attention_mask
246
 
247
+ model_location = "./MM_FT_C1_V2"
248
  # print("Model location:", model_location)
249
 
250
  model = MultimodalPhiModel.from_pretrained(model_location).to(device)