Update app.py

app.py

@@ -20,9 +20,13 @@ def predict(text):
     # Return the label
     return "Image Intent" if predicted_class == 1 else "Non-Image Intent"
 
+# Function to display user history
+def history_display(history):
+    return "\n".join(f"{data['input']} -> {data['output']}" for data in history)
+
 # Define the Gradio interface
 iface = gr.Interface(
-    fn=predict, 
+    fn=predict,
     inputs="text",
     outputs="text",
     live=True,  # Enable user history functionality
@@ -35,7 +39,11 @@ iface = gr.Interface(
         ["Show me cute cat pictures!"],
     ],
     theme="huggingface",  # Dark mode theme
+    allow_flagging=False,  # Disable flagging option
+    live_show_history=True,  # Show user history
+    live_history=True,  # Enable user history recording
+    history=history_display,  # Function to display user history
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()