Update app.py

app.py

@@ -60,6 +60,8 @@ def predict(user_input, history, subject, model, max_tokens, temperature, top_p)
         yield full_message
 
 def generate_image(prompt, size="256x256"):
+    
+    client = OpenAI(api_key=os.getenv("API_TOKEN"))
     try:
         response = client.images.generate(
             model="dall-e-3",  # or "dall-e-2"