foxtrot126 commited on
Commit
6c7a133
·
verified ·
1 Parent(s): 507c0af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -35,8 +35,7 @@ def process_image(image, prompt):
35
 
36
  # Define model 2
37
  processor2 = AutoProcessor.from_pretrained("HuggingFaceM4/Idefics3-8B-Llama3")
38
- model2 = Idefics3ForConditionalGeneration.from_pretrained("HuggingFaceM4/Idefics3-8B-Llama3")
39
- model2.load_adapter("merve/idefics3llama-vqav2").to("cuda")
40
  device2 = infer_device()
41
 
42
  # Define inference function
 
35
 
36
  # Define model 2
37
  processor2 = AutoProcessor.from_pretrained("HuggingFaceM4/Idefics3-8B-Llama3")
38
+ model2 = AutoModel.from_pretrained("merve/idefics3-llama-vqav2", dtype="auto")
 
39
  device2 = infer_device()
40
 
41
  # Define inference function