Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 102 |
print(f"✅ Using device: {device}")
|
| 103 |
|
| 104 |
print("Loading model and tokenizer...")
|
| 105 |
-
model_name = "
|
| 106 |
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
| 107 |
|
| 108 |
model = AutoModel.from_pretrained(
|
|
|
|
| 102 |
print(f"✅ Using device: {device}")
|
| 103 |
|
| 104 |
print("Loading model and tokenizer...")
|
| 105 |
+
model_name = "prithivMLmods/DeepSeek-OCR-Latest-BF16.I64" # - (https://huggingface.co/deepseek-ai/DeepSeek-OCR)
|
| 106 |
tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
| 107 |
|
| 108 |
model = AutoModel.from_pretrained(
|