richardkimsm89 commited on
Commit
65ad07f
·
verified ·
1 Parent(s): 2acd9b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,7 +3,8 @@
3
  import gradio as gr
4
  from huggingface_hub import InferenceClient
5
 
6
- model = "google/gemma-2-27b-it"
 
7
  client = InferenceClient(model)
8
 
9
  def fn(
@@ -43,7 +44,7 @@ def fn(
43
  app = gr.ChatInterface(
44
  fn = fn,
45
  additional_inputs = [
46
- #gr.Textbox(value="You are a friendly Chatbot.", label="System Message"),
47
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens"),
48
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
49
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P"),
 
3
  import gradio as gr
4
  from huggingface_hub import InferenceClient
5
 
6
+ #model = "google/gemma-2-27b-it"
7
+ model = "google/gemma-2-9b-it"
8
  client = InferenceClient(model)
9
 
10
  def fn(
 
44
  app = gr.ChatInterface(
45
  fn = fn,
46
  additional_inputs = [
47
+ #gr.Textbox(value="You are a helpful assistant.", label="System Message"),
48
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens"),
49
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
50
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P"),