nzhabchikov commited on
Commit
b729344
·
1 Parent(s): 314c413

fix gradio

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  import torch
3
  from transformers import AutoProcessor, AutoModelForVision2Seq
4
 
5
- from web_data import custom_css, header, footer, shortcuts
6
 
7
  processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM-256M-Instruct")
8
  model = AutoModelForVision2Seq.from_pretrained(
@@ -11,7 +11,7 @@ model = AutoModelForVision2Seq.from_pretrained(
11
  )
12
 
13
 
14
- def respond(message, history:list[tuple[str, str]], image, system_message):
15
  messages = [
16
  {
17
  "role": "system",
 
2
  import torch
3
  from transformers import AutoProcessor, AutoModelForVision2Seq
4
 
5
+ from web_data import custom_css, header, footer
6
 
7
  processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM-256M-Instruct")
8
  model = AutoModelForVision2Seq.from_pretrained(
 
11
  )
12
 
13
 
14
+ def respond(message, history: list[tuple[str, str]], image, system_message):
15
  messages = [
16
  {
17
  "role": "system",
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  gradio==5.25.0
2
  transformers==4.51.2
3
- torch==2.6.0
 
 
1
  gradio==5.25.0
2
  transformers==4.51.2
3
+ torch==2.6.0
4
+ pydantic==2.10.6