Added empty fig

app.py

@@ -1,6 +1,7 @@
 import os
 import duckdb
 import gradio as gr
+import matplotlib.pyplot as plt
 from transformers import HfEngine, ReactCodeAgent
 from transformers.agents import Tool
 
@@ -65,6 +66,7 @@ def get_visualization(question, tool):
     
     return fig
 
+
 class SQLExecutorTool(Tool):
     name = "sql_engine"
     inputs = {
@@ -83,15 +85,21 @@ class SQLExecutorTool(Tool):
 tool = SQLExecutorTool()
 
 def main(table, text_query):
+    # Empty Fig
+    fig, ax = plt.subplots()
+    ax.set_axis_off()  
+    
     schema, _ = get_table_schema(table)
     tool.description = f"""Allows you to perform SQL queries on the table. Returns a pandas dataframe representation of the result.
     The table schema is as follows: \n{schema}"""
     
+    
     try:
         fig = get_visualization(question=text_query, tool=tool)
     except Exception as e:
         gr.Warning(f"❌ Unable to generate the visualization. {e}")
         
+        
     return fig
     
 

requirements.txt

@@ -1,3 +1,4 @@
+torch
 huggingface_hub
 transformers
 duckdb