Update app.py

app.py

@@ -35,7 +35,7 @@ def get_gpt_response(query):
     inputs = tokenizer(query, return_tensors="pt", padding=True, truncation=True)
     outputs = model(**inputs)
     prediction = torch.argmax(outputs.logits, dim=1)
-    labels = data.columns
+    labels = df.columns
     predicted_label = labels[prediction]
     return predicted_label