sifujohn commited on
Commit
108e847
1 Parent(s): 66bfa19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,10 +1,7 @@
1
  from huggingface_hub import InferenceClient
2
  import gradio as gr
3
 
4
- client = InferenceClient(
5
- "models/google/gemma-7b-it"
6
- )
7
-
8
 
9
  def format_prompt(message, history):
10
  prompt = "<s>"
@@ -97,7 +94,7 @@ gr.ChatInterface(
97
  fn=generate,
98
  chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
99
  additional_inputs=additional_inputs,
100
- title="Gemma 7b",
101
  examples=examples,
102
  concurrency_limit=20,
103
  ).launch(show_api=False)
 
1
  from huggingface_hub import InferenceClient
2
  import gradio as gr
3
 
4
+ gr.load("models/google/gemma-7b")
 
 
 
5
 
6
  def format_prompt(message, history):
7
  prompt = "<s>"
 
94
  fn=generate,
95
  chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
96
  additional_inputs=additional_inputs,
97
+ title="Gemma 7b Chat",
98
  examples=examples,
99
  concurrency_limit=20,
100
  ).launch(show_api=False)