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,
|