Update app.py

app.py

@@ -21,12 +21,11 @@ def answer_question(question,api_key,graph):
     kg = graph[0]
   return kg.chat_qa(question)
 
-def clear_graph(api_key,graph,output_image):
+def clear_graph(api_key,graph):
   graph = []
   kg = KnowledgeGraph(api_key)
   graph.append(kg)
-  w, h = output_image.size
-  return graph,Image.new('RGB', (w, h),(255, 255, 255))
+  return graph,Image.new('RGB', (400, 100),(255, 255, 255))
 
 title = "GPT-4 Knowledge Graph Generator"
 description = "Enter text to generate a knowledge graph using GPT4KG:"
@@ -46,6 +45,6 @@ with gr.Blocks() as demo:
   qa_btn.click(fn=answer_question, inputs=[question,api_key,graph], outputs=[answer])
 
   clear_btn = gr.Button("Clear graph")
-  clear_btn.click(fn=clear_graph, inputs=[api_key,graph,output_image], outputs=[graph,output_image],api_name="clear")
+  clear_btn.click(fn=clear_graph, inputs=[api_key,graph], outputs=[graph,output_image],api_name="clear")
 
 demo.launch()