CreitinGameplays commited on
Commit
d5724ff
·
verified ·
1 Parent(s): 408bcd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -3,18 +3,17 @@ import gradio as gr
3
  # Load the BLOOM model
4
  model = gr.load("models/CreitinGameplays/bloom-3b-conversational")
5
 
6
- # Define the interface with max_length limit
7
  interface = gr.Interface(
8
  fn=model,
9
  inputs="text",
10
- outputs="text",
11
- max_new_tokens=250, # Limit input text length to 250 characters
12
  css="""
13
  .gr-form textarea {
14
  height: 100px; /* Adjust height as needed */
15
  }
16
  """, # Optional: Adjust input text area height
17
- description="Interact with BLOOM (limited to 250 characters)",
18
  )
19
 
20
  # Launch the Gradio interface
 
3
  # Load the BLOOM model
4
  model = gr.load("models/CreitinGameplays/bloom-3b-conversational")
5
 
6
+ # Define the interface
7
  interface = gr.Interface(
8
  fn=model,
9
  inputs="text",
10
+ outputs="text",
 
11
  css="""
12
  .gr-form textarea {
13
  height: 100px; /* Adjust height as needed */
14
  }
15
  """, # Optional: Adjust input text area height
16
+ description="Interact with BLOOM",
17
  )
18
 
19
  # Launch the Gradio interface