Update app.py

app.py

@@ -28,7 +28,10 @@ def generate_image_model_1(prompt, negative_prompt):
 
 def generate_image_model_2(prompt, negative_prompt):
     prompt += " 8k"
-    return model_2(prompt)
+    result = model_2(prompt)
+    if isinstance(result, tuple):
+        return result[0]
+    return result
 
 def generate_image_model_3(prompt, negative_prompt):
     prompt += " 10k"