dexter2389 commited on
Commit
a96199a
·
1 Parent(s): b8b87a2

Remove vlm models

Browse files
Files changed (2) hide show
  1. app.py +0 -6
  2. 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>