Riddhi Bhagwat commited on
Commit
1aaa40e
·
1 Parent(s): 0d9a6c9

changed base model to gemma 3

Browse files
Files changed (2) hide show
  1. app/app.py +4 -4
  2. app/tempCodeRunnerFile.py +1 -0
app/app.py CHANGED
@@ -12,13 +12,13 @@ import spaces
12
  import spaces
13
  import gradio as gr
14
  from feedback import save_feedback, scheduler
15
- from gradio.components.chatbot import Option
16
  from huggingface_hub import InferenceClient
17
  from pandas import DataFrame
18
  from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
19
 
20
 
21
- BASE_MODEL = os.getenv("MODEL", "CohereForAI/aya-expanse-8b")
22
  ZERO_GPU = (
23
  bool(os.getenv("ZERO_GPU", False)) or True
24
  if str(os.getenv("ZERO_GPU")).lower() == "true"
@@ -438,8 +438,8 @@ def wrangle_edit_data(
438
  )
439
  history = history[: index + 1]
440
  history[-1]["options"] = [
441
- Option(label="chosen", value=x.value),
442
- Option(label="rejected", value=original_message["content"]),
443
  ]
444
  return history
445
 
 
12
  import spaces
13
  import gradio as gr
14
  from feedback import save_feedback, scheduler
15
+ from gradio.components.chatbot import OptionDict
16
  from huggingface_hub import InferenceClient
17
  from pandas import DataFrame
18
  from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
19
 
20
 
21
+ BASE_MODEL = os.getenv("MODEL", "google/gemma-3-12b-it")
22
  ZERO_GPU = (
23
  bool(os.getenv("ZERO_GPU", False)) or True
24
  if str(os.getenv("ZERO_GPU")).lower() == "true"
 
438
  )
439
  history = history[: index + 1]
440
  history[-1]["options"] = [
441
+ OptionDict(label="chosen", value=x.value),
442
+ OptionDict(label="rejected", value=original_message["content"]),
443
  ]
444
  return history
445
 
app/tempCodeRunnerFile.py ADDED
@@ -0,0 +1 @@
 
 
1
+ th gr.Blocks(css=cs