i

app.py

@@ -1,3 +1,9 @@
+import torch
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
+
 import gradio as gr
 
 def greet(name):

requirements.txt

@@ -1,10 +1,11 @@
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch
 diffusers==0.27.2
 transformers==4.36.2
 opencv-python
 safetensors
 pillow==10.2.0
 einops
-torch
 peft
 basicsr
 gradio==3.41.2