Fix async

app.py

@@ -7,13 +7,8 @@ executor = ThreadPoolExecutor()
 
 model_pipeline = pipeline("text2text-generation", model="tribler/dsi-search-on-toy-dataset")
 
-async def async_model_pipeline(query):
-    loop = asyncio.get_running_loop()
-    results = await loop.run_in_executor(executor, model_pipeline, query, {"max_length": 60})
-    return results
-
-async def process_query(query):
-    results = await async_model_pipeline(query)
+async def async_process_query(query):
+    results = model_pipeline(query)
     result_text = results[0]['generated_text'].strip()
     if result_text.startswith("http"):
         youtube_id = result_text.split('watch?v=')[-1]
@@ -25,6 +20,13 @@ async def process_query(query):
         bitcoin_logo_url = "https://upload.wikimedia.org/wikipedia/commons/thumb/4/46/Bitcoin.svg/800px-Bitcoin.svg.png"
         return gr.Textbox(f'<div style="display:flex;align-items:center;"><img src="{bitcoin_logo_url}" alt="Bitcoin Logo" style="width:20px;height:20px;margin-right:5px;"><span>{result_text}</span></div>')
 
+
+async def process_query(query):
+    try:
+        return await asyncio.wait_for(async_process_query(query), timeout=10)
+    except asyncio.TimeoutError:
+        return "Processing timed out."
+
 interface = gr.Interface(fn=process_query,
                           inputs=gr.Textbox(label="Query"),
                           outputs="html",