Update main.py

main.py

@@ -48,12 +48,16 @@ async def parse_raw(request: Request):
 # Generate text using the Inference API
 def generate_text(input_text: str) -> str:
     data = {"inputs": input_text}
-    response = requests.post(endpoint, json=data, headers=headers)
+    try:
+        response = requests.post(endpoint, json=data, headers=headers)
+        response.raise_for_status()  # Raise an exception for non-2xx status codes
 
-    print(response.json())  # Print the entire response object
+        print(response.json())  # Print the entire response object
 
-    try:
         return response.json()["generated_text"]
+    except requests.exceptions.RequestException as e:
+        print(f"RequestException: {e}")
+        return ""
     except KeyError as e:
         print(f"KeyError: {e}. Check the response object for the correct key.")
         return ""