Hugging Face's logo

Spaces:
AI-Safeguard
/
Ivy-VL
Running

App Files Community
Ivy1997 commited on
Commit
408282e
•
1 Parent(s): dd3b350

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ import warnings
10
 
11
  warnings.filterwarnings("ignore")
12
 
13
- pretrained = "/tmp/pre-trained/AI-Safeguard/Ivy-VL-llava"
14
  model_name = "llava_qwen"
15
- device = "cuda"
16
  device_map = "auto"
17
 
18
  # Load model, tokenizer, and image processor
 
10
 
11
  warnings.filterwarnings("ignore")
12
 
13
+ pretrained = "AI-Safeguard/Ivy-VL-llava"
14
  model_name = "llava_qwen"
15
+ device = "cpu"
16
  device_map = "auto"
17
 
18
  # Load model, tokenizer, and image processor