Update main.py

main.py

@@ -92,6 +92,7 @@ async def core():
 
 def getPrompt(prompt, modelID, attempts=1):
     response = {}
+    print(modelID)
     if modelID != magic_prompt_model:
         chat = [
             {"role": "user", "content": prompt_base},
@@ -105,9 +106,8 @@ def getPrompt(prompt, modelID, attempts=1):
             if attempts < 3:
                 getPrompt(prompt, modelID, attempts + 1) 
     else:
-        try:
-            print(modelID)
-            apiData={"inputs":input, "parameters": parameters, "options": options, "timeout": 45}
+        try:    
+            apiData={"inputs":prompt, "parameters": parameters, "options": options, "timeout": 45}
             response = requests.post(API_URL + modelID, headers=headers, data=json.dumps(apiData))
             if response.status_code == 200:
                 try: