Update app.py

app.py

@@ -18,14 +18,14 @@ upscaler.enable_xformers_memory_efficient_attention()
 
 def genie (Prompt, negative_prompt, height, width, scale, steps, seed, upscale):
     
-    generator = torch.Generator(device).manual_seed(seed)
-    np.random.seed(0)
+    generator = torch.manual_seed(seed)
+    
     if upscale == "Yes":
-        image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
-        upscaled = upscaler(Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=5, guidance_scale=0, generator=generator).images[0]
+        image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale).images[0]
+        upscaled = upscaler(Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=5, guidance_scale=0).images[0]
         return (image, upscaled)
     else:
-        image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
+        image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale).images[0]
     return (image, image)
     
 gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'),