Update app.py

app.py

@@ -4,20 +4,19 @@ import os
 import spaces
 
 @spaces.GPU
-def query_engine(llm_model, document):
+def query_engine(llm_model, document, query):
     api_key = os.getenv("HUGGINGFACE_API_KEY")
     llm_api_base = "https://api-inference.huggingface.co/models/"
 
-
-    content = document.value.decode("utf-8")
-    docs.append(content)
+    content = str(document)
 
     query_engine = AutoQueryEngine.from_defaults(
-        documents=document,
+        documents=[content],
         llm_model=llm_model,
         llm_api_base=llm_api_base,
     )
-    return query_engine
+    response = query_engine.query(query)
+    return response
 
 interface = gr.Blocks()
 with interface:
@@ -27,5 +26,5 @@ with interface:
     query = gr.Textbox(label="Query")
     output = gr.Textbox(label="Output")
     query_btn = gr.Button("Query")
-    query_btn.click(fn=query_engine, inputs=[llm_model, document], outputs=output)
+    query_btn.click(fn=query_engine, inputs=[llm_model, document, query], outputs=output)
 interface.launch()