fixed output?

app.py

@@ -6,6 +6,7 @@ import os
 import re
 import threading
 import queue
+import time
 
 zero = torch.Tensor([0]).cuda()
 print(zero.device)  # <-- 'cpu' 🤔
@@ -22,17 +23,15 @@ def run_evaluation(model_name):
     results = []
     manifest_logs = []
 
-    # Use the secret HF token from the Hugging Face space
     if "HF_TOKEN" not in os.environ:
         return "Error: HF_TOKEN not found in environment variables.", "Error: Cannot start manifest server without HF_TOKEN."
 
     manifest_process = None
     log_queue = queue.Queue()
     try:
-        # Start manifest server in background with explicit CUDA_VISIBLE_DEVICES
         manifest_cmd = f"""
         cd duckdb-nsql/ && 
-        CUDA_VISIBLE_DEVICES=0 HF_TOKEN={os.environ['HF_TOKEN']}  python -m manifest.api.app \
+        CUDA_VISIBLE_DEVICES=0 HF_TOKEN={os.environ['HF_TOKEN']} python -m manifest.api.app \
         --model_type huggingface \
         --model_generation_type text-generation \
         --model_name_or_path {model_name} \
@@ -46,15 +45,20 @@ def run_evaluation(model_name):
         results.append("Started manifest server in background.")
 
         # Wait for the server to initialize (adjust time as needed)
-        for _ in range(30):
+        server_ready = False
+        for _ in range(60):  # Increased timeout to 60 seconds
             try:
                 line = log_queue.get(timeout=1)
                 manifest_logs.append(line)
                 if "Running on" in line:  # Server is ready
+                    server_ready = True
                     break
             except queue.Empty:
                 pass
 
+        if not server_ready:
+            raise Exception("Manifest server failed to start within the expected time.")
+
         # Run inference
         inference_cmd = f"""
         cd duckdb-nsql/ &&