Upload 29 files

main.py

@@ -119,6 +119,7 @@ def getPrompt(prompt, modelID, attempts=1):
 
 @app.post("/inferencePrompt")
 def inferencePrompt(item: Core):
+    print("Start API Inference Prompt")
     try:
         plain_response_data = getPrompt(item.itemString, prompt_model)
         magic_response_data = getPrompt(item.itemString, magic_prompt_model)
@@ -240,6 +241,7 @@ def inferenceAPI(model, item, attempts = 1):
         response = requests.request("POST", API_URL + model, headers=headers, data=api_data)
         if response is None:
             inferenceAPI(get_random_model(activeModels['text-to-image']), item, attempts+1) 
+        print(response.content[0:200])
         image_stream = BytesIO(response.content)
         image = Image.open(image_stream)
         image.save("response.png")
@@ -304,6 +306,7 @@ def nsfw_check(attempts = 1):
     
 @app.post("/api")
 async def inference(item: Item):
+    print("Start API Inference")
     activeModels = InferenceClient().list_deployed_models()
     base64_img = ""
     model = item.modelID