edgilr commited on
Commit
bab8245
1 Parent(s): 235c21a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,8 +4,7 @@ import gradio as gr
4
  translator = pipeline('text2text-generation', model='edgilr/mbart-neutralization')
5
 
6
  def predict(txt):
7
- return classifier(txt)[0]['label']
8
 
9
  gr.Interface(fn=predict, inputs="text", outputs="text",
10
- examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',
11
- 'the thing looks like a made-for-home-video quickie .']).launch(share=True)
 
4
  translator = pipeline('text2text-generation', model='edgilr/mbart-neutralization')
5
 
6
  def predict(txt):
7
+ return classifier(txt)[0]['generated_text']
8
 
9
  gr.Interface(fn=predict, inputs="text", outputs="text",
10
+ examples=['You are an AI assistant. User will you give you a task. Your goal is to complete the task as faithfully as you can. While performing the task think step-by-step and justify your steps.']).launch(share=True)