CreitinGameplays commited on
Commit
2a472e3
·
verified ·
1 Parent(s): c2eeba9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,7 @@ model = gr.load("models/CreitinGameplays/bloom-3b-conversational")
7
  interface = gr.Interface(
8
  fn=model,
9
  inputs=[
10
- gr.Textbox(label="Text Prompt", value="<|system|> You are a helpful AI assistant </s> <|prompter|> What is an AI? </s> <|assistant|>"),
11
- gr.Slider(minimum=1, maximum=256, label="Max New Tokens", value=128),
12
  ],
13
  outputs="text",
14
  css="""
@@ -18,6 +17,7 @@ interface = gr.Interface(
18
  """, # Optional: Adjust input text area height
19
  description="Interact with BLOOM",
20
  )
 
21
 
22
  # Launch the Gradio interface
23
  interface.launch()
 
7
  interface = gr.Interface(
8
  fn=model,
9
  inputs=[
10
+ gr.Textbox(label="Text Prompt", value="<|system|> You are a helpful AI assistant </s> <|prompter|> What is an AI? </s> <|assistant|>")
 
11
  ],
12
  outputs="text",
13
  css="""
 
17
  """, # Optional: Adjust input text area height
18
  description="Interact with BLOOM",
19
  )
20
+ #max_token = gr.Slider(minimum=1, maximum=256, label="Max New Tokens", value=128),
21
 
22
  # Launch the Gradio interface
23
  interface.launch()