Spaces:
Running
on
Zero
Running
on
Zero
Riddhi Bhagwat
commited on
Commit
·
1aaa40e
1
Parent(s):
0d9a6c9
changed base model to gemma 3
Browse files- app/app.py +4 -4
- 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
|
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", "
|
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 |
-
|
442 |
-
|
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
|