samunder12 commited on
Commit
64f07e3
·
verified ·
1 Parent(s): 3dce9ee

hello again

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
- client = InferenceClient("KoboldAI/fairseq-dense-13B-Shinen")
5
 
6
  def respond(
7
  message,
@@ -34,8 +34,8 @@ def respond(
34
  demo = gr.ChatInterface(
35
  respond,
36
  additional_inputs=[
37
- gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
38
- gr.Slider(minimum=1, maximum=512, value=256, step=1, label="Max new tokens"),
39
  gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
40
  gr.Slider(
41
  minimum=0.1,
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
+ client = InferenceClient("facebook/opt-350m")
5
 
6
  def respond(
7
  message,
 
34
  demo = gr.ChatInterface(
35
  respond,
36
  additional_inputs=[
37
+ gr.Textbox(value="You are a helpful AI assistant.", label="System message"),
38
+ gr.Slider(minimum=1, maximum=256, value=128, step=1, label="Max new tokens"),
39
  gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="Temperature"),
40
  gr.Slider(
41
  minimum=0.1,