Spaces:

maum-ai
/

CANVAS-DEMO

Running on CPU Upgrade

lastdefiance20 commited on Dec 4, 2024

Commit

8e300a7

1 Parent(s): 2bab761

Fix model path

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,7 @@ from canvas import Idefics2Pipeline
 def run_canvas(front_view, map_view, prompt):
     pipeline = Idefics2Pipeline.from_pretrained(
-        # "maum-ai/CANVAS-S"
-        "/jyjung/CANVAS-S"
     )
     messages = [
         {"role": "system", "content": [{"type": "text", "text": prompt}]},

 def run_canvas(front_view, map_view, prompt):
     pipeline = Idefics2Pipeline.from_pretrained(
+        "maum-ai/CANVAS-S"
     )
     messages = [
         {"role": "system", "content": [{"type": "text", "text": prompt}]},