response.content format

app.py

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