Upload app.py

app.py

@@ -20,13 +20,26 @@ def process_data(input_dataframe):
       wineq_t = torch.from_numpy(wineq_np)
       return forward(model, wineq_t)
     return "Invalid input type"
-    
+
+columns = ['fixed acidity',
+ 'volatile acidity',
+ 'citric acid',
+ 'residual sugar',
+ 'chlorides',
+ 'free sulfur dioxide',
+ 'total sulfur dioxide',
+ 'density',
+ 'pH',
+ 'sulphates',
+ 'alcohol']
+
+
 with gr.Blocks() as demo:
     gr.Markdown("Enter your wine data below:")
     input_df = gr.Dataframe(
         row_count=(1, "dynamic"),  # Allows adding/removing rows
         col_count=(11, "dynamic"),  # Allows adding/removing columns
-        headers=list(df.columns)[:-1],
+        headers=columns,
         label="Input Data",
         interactive=True,
         type="pandas" # Specify the desired input type for your function