update cog.yaml

cog.yaml

@@ -10,9 +10,6 @@ build:
     - wget
     - git
     - curl
-    - pytorch
-    - pip=22.2.1
-    - cudatoolkit
     - libgl1-mesa-glx
     - libglib2.0-0
 
@@ -61,8 +58,7 @@ build:
 
   # commands run after the environment is setup
   run:
-    - pip install Pillow==9.5.0
-    - pip install gradio==3.50.0
+    - pip install torch==2.0.1 torchvision==0.15.2
     - "echo env is ready!"
     - "echo another command if needed"