Skip to content

Commit 1f5d0cd

Browse files
committed
changed base model to gemma 3
1 parent 2d93850 commit 1f5d0cd

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

app/app.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,13 +12,13 @@
1212
import spaces
1313
import gradio as gr
1414
from feedback import save_feedback, scheduler
15-
from gradio.components.chatbot import Option
15+
from gradio.components.chatbot import OptionDict
1616
from huggingface_hub import InferenceClient
1717
from pandas import DataFrame
1818
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
1919

2020

21-
BASE_MODEL = os.getenv("MODEL", "CohereForAI/aya-expanse-8b")
21+
BASE_MODEL = os.getenv("MODEL", "google/gemma-3-12b-it")
2222
ZERO_GPU = (
2323
bool(os.getenv("ZERO_GPU", False)) or True
2424
if str(os.getenv("ZERO_GPU")).lower() == "true"
@@ -438,8 +438,8 @@ def wrangle_edit_data(
438438
)
439439
history = history[: index + 1]
440440
history[-1]["options"] = [
441-
Option(label="chosen", value=x.value),
442-
Option(label="rejected", value=original_message["content"]),
441+
OptionDict(label="chosen", value=x.value),
442+
OptionDict(label="rejected", value=original_message["content"]),
443443
]
444444
return history
445445

app/tempCodeRunnerFile.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
th gr.Blocks(css=cs

0 commit comments

Comments
 (0)