Update main.py
Browse files
main.py
CHANGED
@@ -68,3 +68,14 @@ def generate_image():
|
|
68 |
response = generate_image_prodia(prompt, model, sampler, seed, neg)
|
69 |
return jsonify({"image": response})
|
70 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
68 |
response = generate_image_prodia(prompt, model, sampler, seed, neg)
|
69 |
return jsonify({"image": response})
|
70 |
|
71 |
+
@app.route("/test", methods=["POST"])
|
72 |
+
def test():
|
73 |
+
data = request.get_json()
|
74 |
+
prompt = data.get("prompt")
|
75 |
+
model = data.get("model")
|
76 |
+
sampler = data.get("sampler")
|
77 |
+
seed = int(data.get("seed"))
|
78 |
+
neg = data.get("neg")
|
79 |
+
|
80 |
+
res = [prompt, \n, model, \n, sampler, \n, seed, \n, neg]
|
81 |
+
|