Update app.py

app.py

@@ -395,9 +395,11 @@ def invoke (prompt, file, history, rag_option, model_option, openai_api_key, k=3
                 #llm = ChatOpenAI(model_name = MODEL_NAME_OAI_ZEICHNEN,  openai_api_key = openai_api_key, temperature=temperature)#, top_p = top_p)  
                 data = {"inputs": prompt}
                 response = requests.post(API_URL, headers=HEADERS, json=data)
-                result = response.content
-                image = Image.open(io.BytesIO(result))
-                history[-1][1] = image
+                result = response.json()["images"][0]
+                image_b64 = (f"data:image/jpeg;base64,{result}")
+                #result = response.content
+                #image = Image.open(io.BytesIO(result))
+                history[-1][1] = image_b64
                 print("history......................")
                 print(history)
                 return history, "Stop: Success"