Spaces:
Runtime error
Runtime error
Commit
·
1d23ed0
1
Parent(s):
1bf81e3
change model
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from huggingface_hub import InferenceClient
|
|
3 |
import spaces
|
4 |
from transformers import pipeline
|
5 |
|
6 |
-
pipe = pipeline("text-
|
7 |
|
8 |
@spaces.GPU
|
9 |
def respond(
|
@@ -22,7 +22,7 @@ def respond(
|
|
22 |
|
23 |
response = ""
|
24 |
|
25 |
-
for message in pipe
|
26 |
messages,
|
27 |
max_tokens=max_tokens,
|
28 |
stream=True,
|
|
|
3 |
import spaces
|
4 |
from transformers import pipeline
|
5 |
|
6 |
+
pipe = pipeline("image-text-to-text", model="HuggingFaceTB/SmolVLM2-2.2B-Instruct")
|
7 |
|
8 |
@spaces.GPU
|
9 |
def respond(
|
|
|
22 |
|
23 |
response = ""
|
24 |
|
25 |
+
for message in pipe(
|
26 |
messages,
|
27 |
max_tokens=max_tokens,
|
28 |
stream=True,
|