CreitinGameplays commited on
Commit
30cbf95
·
verified ·
1 Parent(s): 4b46184

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -6,9 +6,11 @@ model = gr.load("models/CreitinGameplays/bloom-3b-conversational")
6
  # Define the interface with default prompt
7
  interface = gr.Interface(
8
  fn=model,
9
- inputs="text",
 
 
 
10
  outputs="text",
11
- inputs_opts={"default": "<|system|> You are a helpful AI assistant. </s> <|prompter|> Who was Nikola Tesla? </s> <|assistant|>"}, # Add default prompt here
12
  css="""
13
  .gr-form textarea {
14
  height: 100px; /* Adjust height as needed */
 
6
  # Define the interface with default prompt
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="""
15
  .gr-form textarea {
16
  height: 100px; /* Adjust height as needed */