Spaces:
Sleeping
Sleeping
update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from transformers import AutoConfig
|
|
| 4 |
from models import CustomClassifier
|
| 5 |
|
| 6 |
MODEL_ID = "yhamidullah/custom-classifier-demo"
|
| 7 |
-
config = AutoConfig.from_pretrained(MODEL_ID)
|
| 8 |
model = CustomClassifier.from_pretrained(MODEL_ID)
|
| 9 |
model.eval()
|
| 10 |
|
|
|
|
| 4 |
from models import CustomClassifier
|
| 5 |
|
| 6 |
MODEL_ID = "yhamidullah/custom-classifier-demo"
|
| 7 |
+
#config = AutoConfig.from_pretrained(MODEL_ID)
|
| 8 |
model = CustomClassifier.from_pretrained(MODEL_ID)
|
| 9 |
model.eval()
|
| 10 |
|