Spaces:

CreatorJarvis
/

FoodExtract-Vision

Running on Zero

CreatorJarvis commited on Jan 30

Commit

fdb4ae4

verified ·

1 Parent(s): 3c69511

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def _make_pipe(model_id: str):
     pipe = pipeline(
         "image-text-to-text",
         model=model_id,
-        device=device_arg,
         dtype=DTYPE,
     )
     model = getattr(pipe, "model", None)

     pipe = pipeline(
         "image-text-to-text",
         model=model_id,
+        device=torch.float16,
         dtype=DTYPE,
     )
     model = getattr(pipe, "model", None)