Spaces:

ninjals
/

FoodExtract-Vision-v1

Sleeping

ninjals commited on Feb 5

Commit

8ef3ae6

verified ·

1 Parent(s): 1fcac58

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ original_pipeline = pipeline(
     "image-text-to-text",
     model=BASE_MODEL_ID,
     dtype=torch.float32,
-    device_map="auto"
 )
 # Load fine-tuned model
@@ -23,7 +23,7 @@ ft_pipe = pipeline(
     "image-text-to-text",
     model=FINE_TUNED_MODEL_ID,
     dtype=torch.float32,
-    device_map="auto"
 )
 def create_message(input_image):

     "image-text-to-text",
     model=BASE_MODEL_ID,
     dtype=torch.float32,
+    device_map="cpu"
 )
 # Load fine-tuned model
     "image-text-to-text",
     model=FINE_TUNED_MODEL_ID,
     dtype=torch.float32,
+    device_map="cpu"
 )
 def create_message(input_image):