CreitinGameplays commited on
Commit
6517e51
1 Parent(s): 95ac312

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,7 +3,6 @@ import gradio as gr
3
  def conversation(prompt="", max_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")
7
  response = "Bloom 3b is currently unavailable. Try again later!"
8
  return response
9
 
@@ -13,7 +12,7 @@ interface = gr.Interface(
13
  gr.Textbox(label="Text Prompt", value="<|system|> You are a helpful AI assistant </s> <|prompter|> What is an AI? </s> <|assistant|>"),
14
  gr.Slider(minimum=1, maximum=1024, label="Max New Tokens", value=128),
15
  ],
16
- # No outputs are specified as the response will be printed in the console.
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
  )
 
3
  def conversation(prompt="", max_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
  response = "Bloom 3b is currently unavailable. Try again later!"
7
  return response
8
 
 
12
  gr.Textbox(label="Text Prompt", value="<|system|> You are a helpful AI assistant </s> <|prompter|> What is an AI? </s> <|assistant|>"),
13
  gr.Slider(minimum=1, maximum=1024, label="Max New Tokens", value=128),
14
  ],
15
+ outputs=gr.Textbox(label="AI Assistant Response"), # Textbox for the response
16
  title="Bloom 3b Conversational Assistant",
17
  description="Talk to Bloom 3b using a text prompt and adjust the maximum number of tokens for response generation.",
18
  )