Update main.py

main.py

@@ -104,12 +104,12 @@ def getPrompt(prompt, modelID, attempts=1):
         else:
             apiData={"inputs":prompt, "parameters": parameters, "options": options, "timeout": 45}
             response = requests.post(API_URL + modelID, headers=headers, data=json.dumps(apiData))
+            return response.json()
     except Exception as e:
         print(f"An error occurred: {e}")
         if attempts < 3:
             getPrompt(prompt, modelID, attempts + 1)
-    print(response.json())
-    return response.json()
+    return response
 
 @app.post("/inferencePrompt")
 def inferencePrompt(item: Core):