h-siyuan commited on
Commit
2392f1e
·
verified ·
1 Parent(s): 0469e08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ MAX_PIXELS = 1344 * 28 * 28
19
 
20
  # Load the model
21
  model = Qwen2VLForConditionalGeneration.from_pretrained(
22
- "/users/difei/siyuan/showui_demo/showui-2b",
23
  torch_dtype=torch.bfloat16,
24
  device_map="auto",
25
  )
 
19
 
20
  # Load the model
21
  model = Qwen2VLForConditionalGeneration.from_pretrained(
22
+ "./showui-2b",
23
  torch_dtype=torch.bfloat16,
24
  device_map="auto",
25
  )