Update app.py

app.py

@@ -118,20 +118,7 @@ def general_predictor(*args):
   return cat_value_dicts[final_colname][result[0]]
 
 # add data labels to replace those lost via star-args
-inputls = []
-for colname in data.columns:
-  # skip last column
-  if colname == final_colname:
-    continue
 
-  # access categories dict if data is categorical
-  # otherwise, just use a number input
-  if colname in cat_value_dicts:
-    radio_options = list(cat_value_dicts[colname].keys())
-    inputls.append(gr.inputs.Dropdown(choices=radio_options, type="value", label=colname))
-  else:
-    # add numerical input
-    inputls.append(gr.inputs.Number(label=colname))
 
 block = gr.Blocks()
 
@@ -139,10 +126,26 @@ with open('info.md') as f:
   with block:
     gr.Markdown(f.readline())
     gr.Markdown('Take the quiz to get a personalized recommendation using AI.')
-    gr.Interface(general_predictor, inputs=inputls, outputs="text", allow_flagging='never')
-
-
-# generate gradio interface
+    
+    inputls = []
+    for colname in data.columns:
+      # skip last column
+      if colname == final_colname:
+        continue
+    
+      # access categories dict if data is categorical
+      # otherwise, just use a number input
+      if colname in cat_value_dicts:
+        radio_options = list(cat_value_dicts[colname].keys())
+        inputls.append(gr.inputs.Dropdown(choices=radio_options, type="value", label=colname))
+      else:
+        # add numerical input
+        inputls.append(gr.inputs.Number(label=colname))
+    
+    output = gr.Text()
+    
+    submit = gr.Button("Submit")
+    submit.click(fn=general_predictor, inputs=inputls, outputs=ouput)
 
 # show the interface
 block.launch()