Update app.py

app.py

@@ -13,6 +13,7 @@ def generate_image():
     prompt = request.form['prompt']
     style = request.form['style']
     ratio = request.form['ratio']
+    model = request.form['model']
 
     imagine = Imagine()
 
@@ -20,7 +21,11 @@ def generate_image():
         img_data = imagine.sdprem(
             prompt=prompt,
             style=Style[style],
-            ratio=Ratio[ratio]
+            ratio=Ratio[ratio],
+            seed=1000,
+            cfg=16,
+            model=Model[model],
+            asbase64=False
         )
     except Exception as e:
         return f"An error occurred while generating the image: {e}"
@@ -47,6 +52,7 @@ def api_generate_image():
     prompt = data['prompt']
     style = data['style']
     ratio = data['ratio']
+    model = data['model']
 
     imagine = Imagine()
 
@@ -54,7 +60,11 @@ def api_generate_image():
         img_data = imagine.sdprem(
             prompt=prompt,
             style=Style[style],
-            ratio=Ratio[ratio]
+            ratio=Ratio[ratio],
+            seed=1000,
+            cfg=16,
+            model=Model[model],
+            asbase64=False
         )
     except Exception as e:
         return jsonify({'error': f"An error occurred while generating the image: {e}"}), 500
@@ -78,4 +88,4 @@ def api_generate_image():
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
-            
+