CreitinGameplays commited on
Commit
d70f45a
1 Parent(s): 66a7808

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
 
3
- def conversation(prompt="", max_new_tokens=128):
4
  # Integrate your Bloom 3b model here to generate response based on prompt and max_tokens
5
  # Replace this with the actual call to your Bloom 3b model
6
  gr.load("models/CreitinGameplays/bloom-3b-conversational")
@@ -15,7 +15,7 @@ interface = gr.Interface(
15
  ],
16
  outputs=gr.Textbox(label="AI Assistant Response"), # Textbox for the response
17
  title="Bloom 3b Conversational Assistant",
18
- description="Talk to Bloom 3b using a text prompt and adjust the maximum number of tokens for response generation.",
19
  )
20
 
21
  interface.launch()
 
1
  import gradio as gr
2
 
3
+ def conversation(prompt="", max_lenght=128):
4
  # Integrate your Bloom 3b model here to generate response based on prompt and max_tokens
5
  # Replace this with the actual call to your Bloom 3b model
6
  gr.load("models/CreitinGameplays/bloom-3b-conversational")
 
15
  ],
16
  outputs=gr.Textbox(label="AI Assistant Response"), # Textbox for the response
17
  title="Bloom 3b Conversational Assistant",
18
+ description="Talk to Bloom 3b using a text prompt and adjust the maximum number tokens for response generation.",
19
  )
20
 
21
  interface.launch()