Spaces:

ardavey
/

Vision-Derm

Sleeping

ardavey commited on 25 days ago

Commit

25835df

verified ·

1 Parent(s): e41eb69

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from PIL import Image
 # Load the model and processor
 model_id = "brucewayne0459/paligemma_derm"
 processor = AutoProcessor.from_pretrained(model_id)
-model = PaliGemmaForConditionalGeneration.from_pretrained(model_id, device_map={"": 0})
 model.eval()
 # Set device

 # Load the model and processor
 model_id = "brucewayne0459/paligemma_derm"
 processor = AutoProcessor.from_pretrained(model_id)
+model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
 model.eval()
 # Set device