chheplo commited on
Commit
38f4825
1 Parent(s): b253b9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = ["meta-llama/Meta-Llama-3.1-8B-Instruct"]
10
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
11
  MODEL = os.environ.get("MODEL_ID")
12
 
13
- TITLE = "<h1><center>Meta-Llama-3.1-8B-Instruct</center></h1>"
14
 
15
  PLACEHOLDER = """
16
  <center>
17
- <p>Hi, I'm Llama. Ask me anything.</p>
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 a helpful assistant",
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
  ),