Update gradio_app.py

gradio_app.py

@@ -1,9 +1,13 @@
 import gradio as gr
 import requests
+import os
 
 # Define the FastAPI endpoint
 API_URL = "http://localhost:8000/predict"  # Update this based on your actual endpoint
 
+flagging_dir = "/app/flagged" 
+os.makedirs(flagging_dir, exist_ok=True)
+
 def predict(input1, input2, input3, input4, input5, input6, input7):
     data = {
         "input1": input1,
@@ -21,7 +25,8 @@ def predict(input1, input2, input3, input4, input5, input6, input7):
 iface = gr.Interface(
     fn=predict,
     inputs=["number","number","number","number","number","number","number"],
-    outputs="json"
+    outputs="json",
+    flagging_dir=flagging_dir
 )
 
 iface.launch()