rando

main.py

@@ -31,7 +31,7 @@ def generate_image(prompt, inference_steps, model):
     #pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
     #pipe = pipe.to("cuda")
     
-    pipeline = DiffusionPipeline.from_pretrained('runwayml/stable-diffusion-v1-5')
+    pipeline = DiffusionPipeline.from_pretrained(str(model))
     #pipeline = pipeline.to("cuda")
     generator = torch.Generator("cpu").manual_seed(0)
     pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
@@ -44,11 +44,12 @@ def generate_image(prompt, inference_steps, model):
     #image = pipe(prompt, num_inference_steps=5).images[0]
     #image = pipe(prompt).images[0]
         
-    #filename = 'ai.jpg'
+
+    filename = str(uuid.uuid4()) + ".jpg"
 
     #print(f"after filename assignment")
 
-    image.save('ai.jpg')
+    image.save(filename)
     
     #print(f"after save")
 
@@ -71,7 +72,7 @@ def generate_image(prompt, inference_steps, model):
     with open("assertion.json", "w") as outfile:
         outfile.write(json_object)
     
-    subprocess.check_output(['./truepic-sign', 'sign', 'ai.jpg', '--profile', 'demo', '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/output.jpg')])
+    subprocess.check_output(['./truepic-sign', 'sign', filename, '--profile', 'demo', '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/output.jpg')])
 
     return {"response": "success"}