Update app.py

app.py

@@ -1,25 +1,22 @@
 import subprocess
 import sys
 
-# Ensure required libraries are installed
+# Function to install a package
 def install(package):
     subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
-# Install torch and torchvision
+# Ensure required libraries are installed
 try:
     import torch
 except ImportError:
-    install("torch==2.0.1+cpu")
-    install("torchvision==0.15.2+cpu")
-    install("-f https://download.pytorch.org/whl/torch_stable.html")
+    install("torch==2.0.1")
+    install("torchvision==0.15.2")
 
-# Install numpy
 try:
     import numpy as np
 except ImportError:
     install("numpy<2")
 
-# Install Pillow
 try:
     from PIL import Image
 except ImportError: