Spaces:
Sleeping
Sleeping
cjell commited on
Commit ·
0a9c1ca
1
Parent(s): 628271f
fixing again
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ from fastapi import FastAPI
|
|
| 2 |
from pydantic import BaseModel
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
-
classifier = pipeline("text-classification", model="
|
| 6 |
|
| 7 |
class Query(BaseModel):
|
| 8 |
text: str
|
|
|
|
| 2 |
from pydantic import BaseModel
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
+
classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
|
| 6 |
|
| 7 |
class Query(BaseModel):
|
| 8 |
text: str
|