ganeshkamath89 commited on
Commit
33d6622
·
verified ·
1 Parent(s): 7ee34df

hardcoding a string in the return to unit test functionality

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  def generate_story(story):
5
  ner = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english")
6
- return str(ner(story))
7
 
8
  demo = gr.Interface (
9
  fn=generate_story,
 
3
 
4
  def generate_story(story):
5
  ner = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english")
6
+ return str(ner("My name is Ganesh Kamath and I work at AMD in Bangalore."))
7
 
8
  demo = gr.Interface (
9
  fn=generate_story,