Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import BertTokenizer, BertForSequenceClassification
 import torch
+import uvicorn # <--- THIS IS THE FIX
 
 # --- Configuration ---
 MODEL_PATH = "./BERT_Bullying_Detector_Model"
@@ -34,7 +35,7 @@ def read_root():
 def predict_toxicity(input_data: TextInput):
     if not model or not tokenizer:
         return {"error": "Model not loaded."}
-
+    
     text = input_data.text
     try:
         encoding = tokenizer.encode_plus(
@@ -46,21 +47,21 @@ def predict_toxicity(input_data: TextInput):
             return_tensors='pt',
             truncation=True
         )
-
+        
         input_ids = encoding['input_ids'].to(device)
         attention_mask = encoding['attention_mask'].to(device)
 
         with torch.no_grad():
             outputs = model(input_ids=input_ids, attention_mask=attention_mask)
             _, prediction = torch.max(outputs.logits, dim=1)
-
+        
         label_map = {0: "Not Bullying", 1: "Bullying"}
         result_label = label_map[prediction.item()]
-
+        
         return {"label": result_label, "score": prediction.item()}
     except Exception as e:
         return {"error": str(e)}
-        import uvicorn
 
+# --- Start the Server ---
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)