Update app.py

app.py

@@ -120,4 +120,5 @@ if __name__ == "__main__":
     # Train the model
     model = train_model(dataset)
 
-    # Create
+    # Create a Gradio interface for prediction
+    gr.Interface(fn=predict, inputs=gr.Textbox(), outputs="text").launch()