zmbfeng commited on
Commit
556b209
·
verified ·
1 Parent(s): 5a521e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
7
  tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
8
  model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
9
 
10
- def create_audio(input_str):
11
  #output_raw= generator(input_str)
12
  """print (output_raw)"""
13
 
@@ -18,7 +18,7 @@ def create_audio(input_str):
18
  return (output_str)
19
 
20
  demo = gr.Interface(
21
- fn=create_audio,
22
  inputs="text",
23
  outputs="text",
24
  title="LLM",
 
7
  tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
8
  model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
9
 
10
+ def create_response(input_str):
11
  #output_raw= generator(input_str)
12
  """print (output_raw)"""
13
 
 
18
  return (output_str)
19
 
20
  demo = gr.Interface(
21
+ fn=create_response,
22
  inputs="text",
23
  outputs="text",
24
  title="LLM",