Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ DESCRIPTION= """
|
|
15 |
|
16 |
# model config
|
17 |
model_name = "google/gemma-3-270m-it"
|
18 |
-
model =
|
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
|
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",
|