Update app.py

app.py

@@ -695,15 +695,14 @@ with EmptyInitWrapper():
     sd = {k.replace("model.", ""): v for k, v in sd.items()}
     result = model.load_state_dict(sd)
     
-@torch.cuda.empty_cache()
-
+#@torch.cuda.empty_cache()
 @spaces.GPU(duration=70)
-@torch.no_grad()
+#@torch.no_grad()
 def generate_image(
     prompt, neg_prompt,num_steps ,width, height, guidance, seed,
     do_img2img, init_image, image2image_strength, resize_img, lora_scale,
     progress=gr.Progress(track_tqdm=True)
-):
+): 
     if seed == 0:
         seed = int(random.random() * 1000000)
         
@@ -753,22 +752,6 @@ def generate_image(
     
     return img, seed
 
-class calculateDuration:
-    def __init__(self, activity_name=""):
-        self.activity_name = activity_name
-
-    def __enter__(self):
-        self.start_time = time.time()
-        return self
-    
-    def __exit__(self, exc_type, exc_value, traceback):
-        self.end_time = time.time()
-        self.elapsed_time = self.end_time - self.start_time
-        if self.activity_name:
-            print(f"Elapsed time for {self.activity_name}: {self.elapsed_time:.6f} seconds")
-        else:
-            print(f"Elapsed time: {self.elapsed_time:.6f} seconds")
-
 def update_selection(evt: gr.SelectData, width, height):
     selected_lora = loras[evt.index]
     new_placeholder = f"Type a prompt for {selected_lora['title']}"
@@ -820,8 +803,6 @@ def run_lora(
     pipe.unload_lora_weights()
     return image, seed  
 
-run_lora.zerogpu = True
-
 css = '''
 #gen_btn{height: 100%}
 #title{text-align: center}