devendergarg14
commited on
Commit
•
378d972
1
Parent(s):
863940e
Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,14 @@
|
|
1 |
from transformers import pipeline
|
2 |
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
from transformers import pipeline
|
2 |
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
|
3 |
+
def query(input_sentence,num,start):
|
4 |
+
string3=[]
|
5 |
+
for i in range(0,num):
|
6 |
+
intial="""These are the few examples of converting original sentences into paraphrased sentences.\n original: Symptoms of influenza include fever and nasal congestion.\n paraphrase: A stuffy nose and elevated temperature are signs you may have the flu.\n original: Maintaining a creative work environment is not only beneficial to employees, but also to company profits.\n paraphrase: Having a fertile work environment can increase productivity and profitability.\n """
|
7 |
+
full_input=intial+"original:"+input_sentence + "\n paraphrase:"+start
|
8 |
+
string1=generator("EleutherAI has", do_sample=True, min_length=len(full_input.split())+80)[0]['generated_text']
|
9 |
+
string2=string1.split('paraphrase:',3)[-1]
|
10 |
+
string3.append(string2.split('.',1)[0]+".")
|
11 |
+
return '\n\n'.join([i for i in string3[0:]])
|
12 |
+
title = "Paraphrasing"
|
13 |
+
description = "Gradio Demo for Paraphrasing"
|
14 |
+
gr.Interface(fn=query, inputs=[gr.inputs.Textbox(lines=4, label="Input Text (Single Sentence)"),gr.inputs.Slider( minimum=1, maximum=10, step=1, default=4, label="Numbers of Outputs"),gr.inputs.Textbox(lines=1, label="Starting Point (optional)")],outputs=["text"],title=title,description=description,enable_queue=True).launch()
|