Deepanshu3012 commited on
Commit
12c0ac0
·
1 Parent(s): 4ddeeb7

Fix torch CPU version for HF Spaces deployment

Browse files
Files changed (2) hide show
  1. app.py +9 -0
  2. requirements.txt +6 -5
app.py CHANGED
@@ -2,6 +2,15 @@
2
  Code Review NLP Assistant — Gradio App
3
  Run with: python app.py
4
  """
 
 
 
 
 
 
 
 
 
5
 
6
  import sys
7
  import os
 
2
  Code Review NLP Assistant — Gradio App
3
  Run with: python app.py
4
  """
5
+ import subprocess
6
+ import sys
7
+
8
+ subprocess.run([
9
+ sys.executable, "-m", "pip", "install",
10
+ "torch==2.2.2",
11
+ "--index-url", "https://download.pytorch.org/whl/cpu",
12
+ "--quiet"
13
+ ], check=False)
14
 
15
  import sys
16
  import os
requirements.txt CHANGED
@@ -1,12 +1,13 @@
1
  transformers==4.40.0
2
- torch==2.2.2
 
3
  tokenizers==0.19.1
4
- gradio==4.31.0
5
- tree-sitter==0.21.3
6
- tree-sitter-python==0.21.0
7
  radon==6.0.1
8
  pyflakes==3.2.0
9
  pandas==2.2.2
10
  plotly==5.22.0
11
  sentencepiece==0.2.0
12
- accelerate==0.30.0
 
 
 
1
  transformers==4.40.0
2
+ torch==2.2.2+cpu
3
+ torchvision==0.17.2+cpu
4
  tokenizers==0.19.1
5
+ gradio==4.44.1
 
 
6
  radon==6.0.1
7
  pyflakes==3.2.0
8
  pandas==2.2.2
9
  plotly==5.22.0
10
  sentencepiece==0.2.0
11
+ accelerate==0.30.0
12
+ huggingface_hub==0.23.4
13
+ jinja2==3.1.3