liudongqing commited on
Commit
d9df1f9
·
1 Parent(s): a77a25f

update the dependencies

Browse files
Files changed (2) hide show
  1. app.py +1 -3
  2. requirements.txt +1 -2
app.py CHANGED
@@ -1,9 +1,8 @@
1
  from transformers import AutoProcessor, AutoModelForPreTraining
2
- import torch
3
  from threading import Thread
4
  import gradio as gr
5
  from gradio import FileData
6
- import spaces
7
 
8
  from zipnn import zipnn_hf
9
 
@@ -15,7 +14,6 @@ model = AutoModelForPreTraining.from_pretrained("royleibov/Llama-3.2-11B-Vision-
15
  processor = AutoProcessor.from_pretrained(model)
16
 
17
 
18
- @spaces.GPU
19
  def score_it(input_img):
20
  image = input_img.convert("RGB").resize((224, 224))
21
 
 
1
  from transformers import AutoProcessor, AutoModelForPreTraining
 
2
  from threading import Thread
3
  import gradio as gr
4
  from gradio import FileData
5
+
6
 
7
  from zipnn import zipnn_hf
8
 
 
14
  processor = AutoProcessor.from_pretrained(model)
15
 
16
 
 
17
  def score_it(input_img):
18
  image = input_img.convert("RGB").resize((224, 224))
19
 
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
- torch
2
- spaces
3
  git+https://github.com/huggingface/transformers.git
4
  accelerate>=0.26.0
5
  zipnn
 
1
+ transformers
 
2
  git+https://github.com/huggingface/transformers.git
3
  accelerate>=0.26.0
4
  zipnn