arshadrana commited on
Commit
fe332c7
·
verified ·
1 Parent(s): e0d8673

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,6 +5,6 @@ app = FastAPI()
5
  model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
6
  print("i am gherer enjoying ")
7
  @app.post("/predict")
8
- async def predict(input_text: str):
9
- prediction = model(input_text)
10
- return {"prediction": prediction}
 
5
  model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
6
  print("i am gherer enjoying ")
7
  @app.post("/predict")
8
+ async def predict():
9
+
10
+ return {"prediction": "i am in lodhran "}