remove progress

app.py

@@ -93,20 +93,20 @@ def validate_dimensions(width, height):
     return True, None
 
 @spaces.GPU()
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4): # , progress=gr.Progress(track_tqdm=True)
     try:
-        progress(0, desc="Starting generation...")
+        # progress(0, desc="Starting generation...")
         
         # Validate that prompt is not empty
         if not prompt or prompt.strip() == "":
             return None, "Please provide a valid prompt."
 
-        progress(0.1, desc="Loading image generation model...")
+        # progress(0.1, desc="Loading image generation model...")
         pipe = get_image_gen_pipeline()
         if pipe is None:
             return None, "Image generation model is unavailable."
         
-        progress(0.2, desc="Validating dimensions...")
+        # progress(0.2, desc="Validating dimensions...")
         is_valid, error_msg = validate_dimensions(width, height)
         if not is_valid:
             return None, error_msg
@@ -114,10 +114,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
         
-        progress(0.3, desc="Setting up generator...")
+        # progress(0.3, desc="Setting up generator...")
         generator = torch.Generator("cuda").manual_seed(seed)  # Explicitly use CUDA generator
         
-        progress(0.4, desc="Generating image...")
+        # progress(0.4, desc="Generating image...")
         with torch.autocast('cuda'):
             image = pipe(
                 prompt=prompt, 
@@ -125,12 +125,12 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
                 height=height,
                 num_inference_steps=num_inference_steps, 
                 generator=generator,
-                guidance_scale=7.5,  # Increased guidance scale
-                max_sequence_length=512 
+                guidance_scale=0.0,  # Increased guidance scale
+                # max_sequence_length=512 
             ).images[0]
 
         torch.cuda.empty_cache()  # Clean up GPU memory after generation
-        progress(1.0, desc="Done!")
+        # progress(1.0, desc="Done!")
         return image, seed
     except Exception as e:
         print(f"Error in infer: {str(e)}")  # Add detailed error logging