Chintan-Shah commited on
Commit
6e386c4
·
verified ·
1 Parent(s): 596d986

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -234,7 +234,7 @@ def getInputs(image_path, question, answer=""):
234
 
235
  return start_input_ids, end_input_ids, image_features, attention_mask
236
 
237
- model_location = "./MM_FT_C1_V2"
238
  print("Model location:", model_location)
239
 
240
  model = MultimodalPhiModel.from_pretrained(model_location).to(device)
 
234
 
235
  return start_input_ids, end_input_ids, image_features, attention_mask
236
 
237
+ model_location = "./MM_FT_C1"
238
  print("Model location:", model_location)
239
 
240
  model = MultimodalPhiModel.from_pretrained(model_location).to(device)