Added Qwen

app.py

@@ -15,15 +15,22 @@ print('Connecting to DB...')
 # Connect to DB
 conn = duckdb.connect(f"md:my_db?motherduck_token={md_token}", read_only=True)
 
-models = ["meta-llama/Meta-Llama-3-70B-Instruct",
+models = ["Qwen/Qwen2.5-72B-Instruct","meta-llama/Meta-Llama-3-70B-Instruct",
           "meta-llama/Llama-3.1-70B-Instruct"]
 
+model_loaded = False 
 for model in models:
-  llm_engine = HfEngine(model=model)
-  model_status = llm_engine.client.get_model_status()
-
-  if model_status.loaded:
-    break
+  try:
+      llm_engine = HfEngine(model=model)
+      info = llm_engine.client.get_endpoint_info()
+      model_loaded = True
+      break
+  except Exception as e:
+      print(f"Error for model {model}: {e}")
+      continue
+
+if not model_loaded:
+    gr.Warning(f"❌ None of the model form {models} are available. {e}")
 
 def get_schemas():
     schemas = conn.execute("""
@@ -163,6 +170,7 @@ def main(table, text_query):
     
     schema, table_name = get_table_schema(table)
     
+    
     try:
         output = get_visualization(question=text_query, tool=tool, schema=schema, table_name=table_name)
         fig = output.get('fig', None)