Spaces:

showlab
/

ShowUI

Running on Zero

h-siyuan commited on 15 days ago

Commit

44422c2

•

1 Parent(s): 48a56a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ MAX_PIXELS = 1344 * 28 * 28
 # Load the model
 model = Qwen2VLForConditionalGeneration.from_pretrained(
-    "./showui-2b",
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )

 # Load the model
 model = Qwen2VLForConditionalGeneration.from_pretrained(
+    "showlab/ShowUI-2B",
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )