Revert response parsing

app.py

@@ -80,8 +80,8 @@ def create_prompt_for_image_generation(user_prompt: str) -> str:
         # response = model(
         #     prompt=prompt, temperature=1., max_tokens=512)
         response = model(messages, stop_sequences=["END"])
-        return response['choices'][0]['text'] 
-        # return response
+        # return response['choices'][0]['text'] 
+        return response
 
     except Exception as e:
         print(f"Error during LLM call: {str(e)}")