Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,7 +64,7 @@ except Exception as e:
|
|
| 64 |
print("Loading tokenizer...")
|
| 65 |
try:
|
| 66 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 67 |
-
MODEL_ID, # Load directly from Hub
|
| 68 |
cache_dir=cache_dir,
|
| 69 |
trust_remote_code=True
|
| 70 |
)
|
|
@@ -133,4 +133,4 @@ torch.cuda.empty_cache()
|
|
| 133 |
|
| 134 |
if __name__ == "__main__":
|
| 135 |
import uvicorn
|
| 136 |
-
uvicorn.run(app, host="0.0.0.0", port=
|
|
|
|
| 64 |
print("Loading tokenizer...")
|
| 65 |
try:
|
| 66 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 67 |
+
MODEL_ID, # Load directly from Hub
|
| 68 |
cache_dir=cache_dir,
|
| 69 |
trust_remote_code=True
|
| 70 |
)
|
|
|
|
| 133 |
|
| 134 |
if __name__ == "__main__":
|
| 135 |
import uvicorn
|
| 136 |
+
uvicorn.run(app, host="0.0.0.0", port=8000)
|