Update main.py

main.py

@@ -7,17 +7,13 @@ app = FastAPI()
 # Define the Gradio API endpoint
 gradio_api_url = "https://astro21-test-2.hf.space/--replicas/x5m8s/"
 
+# Function to summarize a file using the Gradio API
 # Function to summarize a file using the Gradio API
 async def summarize_file_with_gradio(file_content: str) -> Dict[str, str]:
     try:
-        async with httpx.AsyncClient() as client:
-            response = await client.post(f"{gradio_api_url}/predict", json={"file_content": file_content})
-            if response.status_code == 200:
-                result = response.json()
-                print(result)
-                return {"summary": result[0]}
-            else:
-                return {"error": "Gradio API request failed"}
+        client = GradioClient(gradio_api_url)
+        result = await client.predict(file_content, api_name="/predict")
+        return {"summary": result[0]}
     except Exception as e:
         return {"error": str(e)}