khalednabawi11 commited on
Commit
bc58772
·
verified ·
1 Parent(s): 6382bc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -71,6 +71,10 @@ processor = AutoProcessor.from_pretrained("google/medgemma-4b-it")
71
  model = AutoModelForImageTextToText.from_pretrained("google/medgemma-4b-it", device_map = "cpu")
72
 
73
 
 
 
 
 
74
  model.eval()
75
 
76
  # Inference function
 
71
  model = AutoModelForImageTextToText.from_pretrained("google/medgemma-4b-it", device_map = "cpu")
72
 
73
 
74
+ processor = AutoProcessor.from_pretrained("google/gemma-3n-E4B-it-litert-preview")
75
+ model = AutoModelForImageTextToText.from_pretrained("google/gemma-3n-E4B-it-litert-preview", device_map = "cpu")
76
+
77
+
78
  model.eval()
79
 
80
  # Inference function