Spaces:
Sleeping
Sleeping
liudongqing
commited on
Commit
·
d9df1f9
1
Parent(s):
a77a25f
update the dependencies
Browse files- app.py +1 -3
- 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 |
-
|
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 |
-
|
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
|