Norod78 commited on
Commit
5b3c141
·
verified ·
1 Parent(s): 0e1bb1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,7 +15,7 @@ DESCRIPTION= """
15
 
16
  # model config
17
  model_name = "google/gemma-3-270m-it"
18
- model = Gemma3ForConditionalGeneration.from_pretrained(
19
  model_name,
20
  torch_dtype="auto",
21
  device_map="auto",
@@ -148,14 +148,14 @@ chat_interface = gr.ChatInterface(
148
  gr.Slider(label="Repetition Penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.1),
149
  ],
150
  examples=[
151
- [{"text": "Write a poem which describes this image", "files": ["examples/image1.jpg"]}],
152
  ],
153
  textbox=gr.MultimodalTextbox(
154
  rtl=False,
155
  label="input",
156
  file_types=["image", "video"],
157
  file_count="multiple",
158
- placeholder="Input text, image or video",
159
  ),
160
  cache_examples=False,
161
  type="messages",
 
15
 
16
  # model config
17
  model_name = "google/gemma-3-270m-it"
18
+ model = AutoModelForCausalLM.from_pretrained(
19
  model_name,
20
  torch_dtype="auto",
21
  device_map="auto",
 
148
  gr.Slider(label="Repetition Penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.1),
149
  ],
150
  examples=[
151
+ [{"text": "Write a poem which describes potatoes"}],
152
  ],
153
  textbox=gr.MultimodalTextbox(
154
  rtl=False,
155
  label="input",
156
  file_types=["image", "video"],
157
  file_count="multiple",
158
+ placeholder="Input text, Any image or video will be ignored",
159
  ),
160
  cache_examples=False,
161
  type="messages",