Spaces:

koyu008
/

Toxic_Comment_Classifier

Running

koyu008 commited on Jun 27

Commit

3a3863a

verified ·

1 Parent(s): 16b2ba7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,3 +102,8 @@ def predict(data: TextIn):
             outputs = model(**inputs)
             probs = torch.softmax(outputs, dim=1).squeeze().cpu().tolist()
         return {"language": "Hinglish", "predictions": dict(zip(hinglish_labels, probs))}

             outputs = model(**inputs)
             probs = torch.softmax(outputs, dim=1).squeeze().cpu().tolist()
         return {"language": "Hinglish", "predictions": dict(zip(hinglish_labels, probs))}
+@app.get("/")
+def root():
+    return {"message": "API is running"}