update

Dockerfile

@@ -24,5 +24,4 @@ COPY --chown=user ./out /app/out
 COPY --chown=user . /app
 
 RUN ls -R /app/out
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
-
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -22,8 +22,6 @@ class PredictionInput(BaseModel):
     text: str = Field(..., title="Text to classify", description="The text to classify for malicious content")
     context: Optional[str] = Field(None, title="Context for classification", description="Optional context to provide additional information for classification")
 
-app.mount("/", StaticFiles(directory="/app/out", html=True), name="static")
-
 @app.post("/predict")
 def predict(input: PredictionInput):
     try:
@@ -35,9 +33,12 @@ def predict(input: PredictionInput):
             raise HTTPException(status_code=400, detail="Context input exceeds maximum length of 512 characters")
         if not input.context:
             result = model.predict(input.text, device="cpu")
+            print(result)
             return {"text": input.text, "prediction": result[0]["prediction"], "probabilities": result[0]["probabilities"]}
         else:
             result = model.predict([[input.text,input.context]], device="cpu")
             return {"text": input.text, "context": input.context, "prediction": result[0]["prediction"], "probabilities": result[0]["probabilities"]}
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+    
+app.mount("/", StaticFiles(directory="out", html=True), name="static")