Spaces:
Sleeping
Sleeping
Commit
·
a96199a
1
Parent(s):
b8b87a2
Remove vlm models
Browse files- app.py +0 -6
- frontend.html +0 -2
app.py
CHANGED
|
@@ -16,15 +16,11 @@ from transformers import pipeline
|
|
| 16 |
|
| 17 |
class SupportedModelPipes(StrEnum):
|
| 18 |
SmolLLM2 = "smollm2"
|
| 19 |
-
SmolVLM = "smolvlm"
|
| 20 |
|
| 21 |
|
| 22 |
smollm2_pipeline = pipeline(
|
| 23 |
"text-generation", model="HuggingFaceTB/SmolLM2-135M-Instruct"
|
| 24 |
)
|
| 25 |
-
smolvlm_pipeline = pipeline(
|
| 26 |
-
"image-text-to-text", model="HuggingFaceTB/SmolVLM-500M-Instruct"
|
| 27 |
-
)
|
| 28 |
|
| 29 |
|
| 30 |
class ChatRequest(BaseModel):
|
|
@@ -70,8 +66,6 @@ def chat(payload: ChatRequest, request: Request):
|
|
| 70 |
match payload.model:
|
| 71 |
case SupportedModelPipes.SmolLLM2:
|
| 72 |
ai_pipeline = smollm2_pipeline
|
| 73 |
-
case SupportedModelPipes.SmolVLM:
|
| 74 |
-
ai_pipeline = smolvlm_pipeline
|
| 75 |
|
| 76 |
ai_response = (
|
| 77 |
ai_pipeline(
|
|
|
|
| 16 |
|
| 17 |
class SupportedModelPipes(StrEnum):
|
| 18 |
SmolLLM2 = "smollm2"
|
|
|
|
| 19 |
|
| 20 |
|
| 21 |
smollm2_pipeline = pipeline(
|
| 22 |
"text-generation", model="HuggingFaceTB/SmolLM2-135M-Instruct"
|
| 23 |
)
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
|
| 26 |
class ChatRequest(BaseModel):
|
|
|
|
| 66 |
match payload.model:
|
| 67 |
case SupportedModelPipes.SmolLLM2:
|
| 68 |
ai_pipeline = smollm2_pipeline
|
|
|
|
|
|
|
| 69 |
|
| 70 |
ai_response = (
|
| 71 |
ai_pipeline(
|
frontend.html
CHANGED
|
@@ -63,8 +63,6 @@
|
|
| 63 |
<ul class="py-1">
|
| 64 |
<li class="model-option px-4 py-2 hover:bg-gray-100 cursor-pointer" data-value="smollm2">
|
| 65 |
SmolLM2</li>
|
| 66 |
-
<li class="model-option px-4 py-2 hover:bg-gray-100 cursor-pointer" data-value="smolvlm">
|
| 67 |
-
SmolVLM</li>
|
| 68 |
</ul>
|
| 69 |
</div>
|
| 70 |
</div>
|
|
|
|
| 63 |
<ul class="py-1">
|
| 64 |
<li class="model-option px-4 py-2 hover:bg-gray-100 cursor-pointer" data-value="smollm2">
|
| 65 |
SmolLM2</li>
|
|
|
|
|
|
|
| 66 |
</ul>
|
| 67 |
</div>
|
| 68 |
</div>
|