Update app.py

app.py

@@ -32,6 +32,7 @@ from models.eva_clip import create_model_and_transforms
 from models.eva_clip.constants import OPENAI_DATASET_MEAN, OPENAI_DATASET_STD
 from models.eva_clip.utils_qformer import resize_numpy_image_long
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
+os.environ["ZERO_GPU_PATCH_TORCH_DEVICE"] = "True"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")
@@ -139,7 +140,7 @@ def delete_old_files():
                     if file_mtime < cutoff:
                         os.remove(file_path)
         time.sleep(600)
-@spaces.GPU(duration=180)
+@spaces.GPU()
 def plex(prompt,image_input,seed_value,scale_status,rife_status,progress=gr.Progress(track_tqdm=True)):
     seed = seed_value
     if seed == -1: