Spaces:

edemana
/

my-stylegan-model

Runtime error

edemana commited on Aug 10, 2024

Commit

cc63fb6

verified ·

1 Parent(s): 445a288

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -9,10 +9,10 @@ RUN apt-get update && apt-get install -y \
 # Install Python dependencies
 RUN pip install --no-cache-dir \
-    torch==2.0.1+cu121 \
-    torchvision==0.15.1+cu121 \
-    torchaudio==2.1.1+cu121 \
-    gradio==3.7.1 \
     tqdm \
     pillow \
     numpy

 # Install Python dependencies
 RUN pip install --no-cache-dir \
+    torch \
+    torchvision \
+    torchaudio \
+    gradio \
     tqdm \
     pillow \
     numpy