Update main.py

main.py

@@ -108,6 +108,7 @@ def getPrompt(prompt, modelID, attempts=1):
         print(f"An error occurred: {e}")
         if attempts < 3:
             getPrompt(prompt, modelID, attempts + 1)
+    print(response.json())
     return response.json()
 
 @app.post("/inferencePrompt")