Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoModel, AutoTokenizer, AutoFeatureExtractor
|
|
3 |
from PIL import Image
|
4 |
|
5 |
# Load Deepseek-vl2-small model and tokenizer
|
6 |
-
model_name = "
|
7 |
model = AutoModel.from_pretrained(model_name)
|
8 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
9 |
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
|
@@ -25,7 +25,7 @@ interface = gr.Interface(
|
|
25 |
fn=process_image_text,
|
26 |
inputs=[gr.Image(type="pil"), gr.Textbox()],
|
27 |
outputs="text",
|
28 |
-
title="
|
29 |
)
|
30 |
|
31 |
# Launch app
|
|
|
3 |
from PIL import Image
|
4 |
|
5 |
# Load Deepseek-vl2-small model and tokenizer
|
6 |
+
model_name = "deepseek-vl2-small" # Replace with actual model name if available on HF
|
7 |
model = AutoModel.from_pretrained(model_name)
|
8 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
9 |
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
|
|
|
25 |
fn=process_image_text,
|
26 |
inputs=[gr.Image(type="pil"), gr.Textbox()],
|
27 |
outputs="text",
|
28 |
+
title="deepseek-vl2-small Demo"
|
29 |
)
|
30 |
|
31 |
# Launch app
|