Update app.py

app.py

@@ -21,7 +21,7 @@ def predict_author(text: str):
     with torch.no_grad():
         outputs = model(**inputs)
         probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
-                pred = torch.argmax(probs, dim=1).item()
+        pred = torch.argmax(probs, dim=1).item()
         confidence = probs[0][pred].item()
     predicted_author = label_map[pred]
     return predicted_author, round(confidence * 100, 2)
@@ -45,4 +45,3 @@ async def whatsapp_reply(Body: str = Form(...)):
 async def predict(text: str):
     author, confidence = predict_author(text)
     return {"prediction": author, "confidence": confidence}
-