Spaces:

Geraldine
/

Image-to-text-SmolVLM-for-Omeka

Sleeping

Geraldine commited on 12 days ago

Commit

4063e57

verified ·

1 Parent(s): 398493c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,12 +12,12 @@ import os, stat, io
 # Load the model in half-precision on the available device(s)
 model = AutoModelForVision2Seq.from_pretrained(
-    "./SmolVLM-500M-Instruct",
     torch_dtype=torch.float32,
     _attn_implementation="eager",
     device_map="cpu"
 )
-processor = AutoProcessor.from_pretrained("./SmolVLM-500M-Instruct")
 def array_to_image(image_array):
     if image_array is None:

 # Load the model in half-precision on the available device(s)
 model = AutoModelForVision2Seq.from_pretrained(
+    "HuggingFaceTB/SmolVLM-500M-Instruct",
     torch_dtype=torch.float32,
     _attn_implementation="eager",
     device_map="cpu"
 )
+processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM-500M-Instruct")
 def array_to_image(image_array):
     if image_array is None: