File tree Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Expand file tree Collapse file tree 2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change 1212import spaces
1313import gradio as gr
1414from feedback import save_feedback , scheduler
15- from gradio .components .chatbot import Option
15+ from gradio .components .chatbot import OptionDict
1616from huggingface_hub import InferenceClient
1717from pandas import DataFrame
1818from 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 " )
2222ZERO_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
Original file line number Diff line number Diff line change 1+ th gr .Blocks (css = cs
You can’t perform that action at this time.
0 commit comments