Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,15 +6,15 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
|
|
6 |
import gradio as gr
|
7 |
from threading import Thread
|
8 |
|
9 |
-
MODEL_LIST = ["
|
10 |
HF_TOKEN = os.environ.get("HF_TOKEN", None)
|
11 |
MODEL = os.environ.get("MODEL_ID")
|
12 |
|
13 |
-
TITLE = "<h1><center>
|
14 |
|
15 |
PLACEHOLDER = """
|
16 |
<center>
|
17 |
-
<p>Hi, I'm
|
18 |
</center>
|
19 |
"""
|
20 |
|
@@ -107,7 +107,7 @@ with gr.Blocks(css=CSS, theme="Nymbo/Alyx_Theme") as demo:
|
|
107 |
additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
|
108 |
additional_inputs=[
|
109 |
gr.Textbox(
|
110 |
-
value="You are
|
111 |
label="System Prompt",
|
112 |
render=False,
|
113 |
),
|
|
|
6 |
import gradio as gr
|
7 |
from threading import Thread
|
8 |
|
9 |
+
MODEL_LIST = ["chheplo/dhenu2-synth1-lora"]
|
10 |
HF_TOKEN = os.environ.get("HF_TOKEN", None)
|
11 |
MODEL = os.environ.get("MODEL_ID")
|
12 |
|
13 |
+
TITLE = "<h1><center>Dhenu2-synth1-llama-3.1-8b-instruct</center></h1>"
|
14 |
|
15 |
PLACEHOLDER = """
|
16 |
<center>
|
17 |
+
<p>Hi, I'm Dhenu. Ask me anything about Agriculture in India.</p>
|
18 |
</center>
|
19 |
"""
|
20 |
|
|
|
107 |
additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
|
108 |
additional_inputs=[
|
109 |
gr.Textbox(
|
110 |
+
value="You are an agriculture assistant.",
|
111 |
label="System Prompt",
|
112 |
render=False,
|
113 |
),
|