Update app.py

app.py

@@ -364,7 +364,8 @@ def umwandeln_fuer_anzeige(image):
 
 def generate_auswahl(prompt, file, history, rag_option, model_option, openai_api_key, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,):
      if (prompt.find('zeichnen') != -1):
-        result = generate_bild(prompt)
+        response = generate_bild(prompt)
+        result = response.content 
         #Bild ausgeben
         image = Image.open(io.BytesIO(result)) 
         
@@ -398,8 +399,7 @@ def generate_bild(prompt):
     data = {"inputs": prompt}
     response = requests.post(API_URL, headers=HEADERS, json=data)
     print("fertig Bild")
-    result = response.content 
-    return result
+    return response
 
 def generate_text (prompt, file, history, rag_option, model_option, openai_api_key, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,):
     global splittet