Spaces:
Runtime error
Runtime error
Commit
·
355b8bb
1
Parent(s):
139dafe
First commit
Browse files- app/model.py +1 -0
app/model.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 2 |
import torch
|
| 3 |
from app.config import MODEL_NAME, DEVICE
|
|
|
|
| 1 |
+
import os
|
| 2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 3 |
import torch
|
| 4 |
from app.config import MODEL_NAME, DEVICE
|