improve submission

app.py

@@ -9,19 +9,26 @@ app = FastAPI(
     title="Frugal AI Challenge API",
     description="API for the Frugal AI Challenge evaluation endpoints"
 )
+from fastapi import FastAPI
+from tasks.text import router as text_router
+
+@app.get("/health")
+async def health_check():
+    return {"status": "ok"}
 
+app.include_router(text_router)
 # Include all routers
-app.include_router(text.router)
-app.include_router(image.router)
-app.include_router(audio.router)
+# app.include_router(text.router)
+# app.include_router(image.router)
+# app.include_router(audio.router)
 
 @app.get("/")
 async def root():
     return {
         "message": "Welcome to the Frugal AI Challenge API",
         "endpoints": {
-            "text": "/text - Text classification task",
-            "image": "/image - Image classification task (coming soon)",
-            "audio": "/audio - Audio classification task (coming soon)"
+            "text": "/text - Text classification task"#,
+#            "image": "/image - Image classification task (coming soon)",
+#            "audio": "/audio - Audio classification task (coming soon)"
         }
     } 

submission_script.py

@@ -14,32 +14,42 @@ def evaluate_text_model(space_url: str, max_retries=3, retry_delay=5):
         "test_seed": 42,
     }
     
-    # Construct API URL
+    # Construct base URL and API endpoints
     if "localhost" in space_url:
-        api_url = f"{space_url}/text"
+        base_url = space_url
     else:
-        api_url = f"https://{space_url.replace('/', '-')}.hf.space/text"
+        base_url = f"https://{space_url.replace('/', '-')}.hf.space"
+    
+    api_url = f"{base_url}/text"
+    health_url = f"{base_url}/health"
 
     headers = {
         'Content-Type': 'application/json',
         'Accept': 'application/json'
     }
 
+    print(f"Base URL: {base_url}")
+    print(f"API URL: {api_url}")
+    print(f"Health check URL: {health_url}")
+
     for attempt in range(max_retries):
         try:
             print(f"\nAttempt {attempt + 1} of {max_retries}")
-            print(f"Making request to: {api_url}")
             
             # Health check
-            health_url = f"https://{space_url.replace('/', '-')}.hf.space/health"
+            print("Checking space health...")
             health_response = requests.get(health_url, timeout=30)
+            print(f"Health check status: {health_response.status_code}")
             
             if health_response.status_code != 200:
                 print(f"Space not ready (status: {health_response.status_code})")
-                time.sleep(retry_delay)
+                if attempt < max_retries - 1:
+                    print(f"Waiting {retry_delay} seconds before retry...")
+                    time.sleep(retry_delay)
                 continue
 
             # Make API call
+            print("Making evaluation request...")
             response = requests.post(
                 api_url,
                 json=params,
@@ -47,20 +57,26 @@ def evaluate_text_model(space_url: str, max_retries=3, retry_delay=5):
                 timeout=300
             )
             
+            print(f"Evaluation response status: {response.status_code}")
+            
             if response.status_code == 200:
                 return response.json()
             else:
                 print(f"Error: Status {response.status_code}")
                 print(f"Response: {response.text}")
-                time.sleep(retry_delay)
+                if attempt < max_retries - 1:
+                    print(f"Waiting {retry_delay} seconds before retry...")
+                    time.sleep(retry_delay)
                 
         except requests.exceptions.RequestException as e:
             print(f"Request error: {str(e)}")
             if attempt < max_retries - 1:
+                print(f"Waiting {retry_delay} seconds before retry...")
                 time.sleep(retry_delay)
         except Exception as e:
             print(f"Unexpected error: {str(e)}")
             if attempt < max_retries - 1:
+                print(f"Waiting {retry_delay} seconds before retry...")
                 time.sleep(retry_delay)
     
     return None