Update app.py

app.py

@@ -1,3 +1,12 @@
+from transformers import pipeline
 import gradio as gr
 
-gr.load("models/Ajay1311/phish").launch()
+# Manually load the model
+model = pipeline('text-classification', model="Ajay1311/phish")
+
+# Define a function for Gradio interface
+def classify(text):
+    return model(text)
+
+# Create the Gradio interface
+gr.Interface(fn=classify, inputs="text", outputs="json").launch()