Update app.py

app.py

@@ -62,11 +62,18 @@ def predict(text):
         outputs = model.predict(inputs)
         print(f"Model outputs: {outputs}")
         
-        return f"This text is a violation = {outputs[0][0]:.2f}"
+        # Interpret the output as a prediction
+        prediction = outputs[0][0]
+        if prediction >= 0.5:
+            result = f"This text is a violation = {prediction:.2f}"
+        else:
+            result = f"This text is not a violation = {prediction:.2f}"
+        
+        return result
     except Exception as e:
         print(f"Error during prediction: {e}")
         return f"Error during prediction: {e}"
 
 # Set up the Gradio interface
 demo = gr.Interface(fn=predict, inputs="text", outputs="text")
-demo.launch()
+demo.launch()