Upload app.py

app.py

@@ -21,6 +21,8 @@ user_agents = dataset['train'].to_pandas()
 user_agents = user_agents.iloc[:,1:]
 user_batch = user_agents[:10]
 
+print(user_batch)
+
 def create_description(row):
     description = (
         f"Imagine that you are currently {int(row['age'])} years old. You have {int(row['num_pets'])} pets "
@@ -83,7 +85,7 @@ def query_agent(description, question, image0, image1):
         print(f"Failed to analyze {image0} and {image1} after 3 attempts.")
         return None, None
 
-def simulate(user_batch, question, image0, image1):
+def simulate(image0, image1):
     preferences = []
     reasons = []
     probs = []
@@ -111,9 +113,7 @@ def simulate(user_batch, question, image0, image1):
 subtitle = "Upload two images of emails and see which is generally preferred by Petco customers!"
 
 demo = gr.Interface(fn=simulate, 
-                    inputs=[user_batch,
-                            question,
-                            gr.UploadButton("Click to Upload Email 0", file_types=["image"], file_count="1"), 
+                    inputs=[gr.UploadButton("Click to Upload Email 0", file_types=["image"], file_count="1"), 
                             gr.UploadButton("Click to Upload Email 1", file_types=["image"], file_count="1")], 
                     outputs="text",
                     title="Pairwise Simulation of Petco Email Preference",