Update app.py

app.py

@@ -10,7 +10,7 @@ import traceback
 
 # Constants
 IMG_SIZE = 128
-TIMESTEPS = 300
+TIMESTEPS = 500
 NUM_CLASSES = 2
 
 # Global Cancellation Flag
@@ -231,7 +231,7 @@ def load_model(model_path, device):
     
     diffusion_model.eval()
     return diffusion_model
-    
+
 MODEL_NAME = "model_weights.pth"
 model_path = MODEL_NAME
 print("Loading model...")
@@ -299,7 +299,7 @@ def generate_images(label_str, num_images, progress=gr.Progress()):
         raise gr.Error(f"Generation failed: {str(e)}")
     finally:
         torch.cuda.empty_cache()
-        
+
 # Gradio UI
 with gr.Blocks(theme=gr.themes.Soft(
     primary_hue="violet",