Spaces:
Runtime error
Runtime error
Pedrinho-Dev01 commited on
Commit ·
56417b0
1
Parent(s): eb496e4
Update
Browse files
api.py
CHANGED
|
@@ -50,7 +50,7 @@ app.add_middleware(
|
|
| 50 |
class ModelBundle:
|
| 51 |
def __init__(self, repo_id: str, model_class):
|
| 52 |
print(f"Loading {repo_id} …")
|
| 53 |
-
self.tokenizer = AutoTokenizer.from_pretrained(repo_id
|
| 54 |
self.model = model_class.from_pretrained(repo_id)
|
| 55 |
self.model.to(DEVICE)
|
| 56 |
self.model.eval()
|
|
|
|
| 50 |
class ModelBundle:
|
| 51 |
def __init__(self, repo_id: str, model_class):
|
| 52 |
print(f"Loading {repo_id} …")
|
| 53 |
+
self.tokenizer = AutoTokenizer.from_pretrained(repo_id)
|
| 54 |
self.model = model_class.from_pretrained(repo_id)
|
| 55 |
self.model.to(DEVICE)
|
| 56 |
self.model.eval()
|