Update src/kandinsky.py

src/kandinsky.py

@@ -35,16 +35,12 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU #[uncomment to use ZeroGPU]
-def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
+def generate_image(prompt, width, height, guidance_scale=3, num_inference_steps=50, progress=gr.Progress(track_tqdm=True)):
+
+    # generator = torch.Generator().manual_seed(0)
 
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-        
-    generator = torch.Generator().manual_seed(seed)
-    
     image = pipe(
         prompt = prompt, 
-        negative_prompt = negative_prompt,
         guidance_scale = guidance_scale, 
         num_inference_steps = num_inference_steps, 
         width = width, 
@@ -52,7 +48,7 @@ def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height,
         generator = generator
     ).images[0] 
     
-    return image, seed
+    return image
 
 
 # device_map = torch.device('cuda:0')