Update app.py

app.py

@@ -82,9 +82,8 @@ else:
     is_count_query = False
 
     # Check if the question contains "graph"
+    
     if 'graph' in question.lower():
-        is_count_query = True
-    elif 'graph' in question.lower():
         is_graph_query = True
 
     # Process the answer using TAPAS and T5
@@ -126,9 +125,8 @@ else:
 
                 else:
                     # Handle graph-related questions
-                    if is_count_query:
-                        # Extract the column name to count
-                        column_name = question.split('count')[-1].strip()
+                    
+                        
 
                         if column_name in df.columns:
                             # Ask TAPAS to count the rows for this specific column