Update app.py

app.py

@@ -11,7 +11,6 @@ import uuid
 import gradio as gr
 import numpy as np
 from PIL import Image
-import torch
 from diffusers import AutoencoderKL, StableDiffusionXLPipeline
 from diffusers import EulerAncestralDiscreteScheduler
 from typing import Tuple
@@ -20,6 +19,10 @@ import datetime
 import cyper
 from image_gen_aux import UpscaleWithModel
 
+from torch_cuda_installer import install_pytorch
+install_pytorch(cuda_key=None, packages=['torch', 'torchvision', 'torchaudio'])
+import torch
+
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
 torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False