Update app.py

app.py

@@ -84,10 +84,10 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
     random.seed(time.time())
     seed = int(random.randrange(4294967294))
     if segment==1:
-        Offload.offload(
-            pipeline=pipe,
-            config=offload_config,
-        )
+        #Offload.offload(
+        #    pipeline=pipe,
+        #    config=offload_config,
+        #)
         prompt_embeds, prompt_attention_mask, negative_prompt_embeds, negative_attention_mask, pooled_prompt_embeds, negative_pooled_prompt_embeds = pipe.encode_prompt(
         prompt=prompt, do_classifier_free_guidance=True, negative_prompt=negative_prompt, device=device
         )