Commit
·
5bae3bb
1
Parent(s):
494e732
Update app.py
Browse files
app.py
CHANGED
@@ -1,13 +1,12 @@
|
|
1 |
from transformers import pipeline
|
2 |
import gradio as gr
|
3 |
-
import random
|
4 |
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
|
5 |
def query(input_sentence,num,start):
|
6 |
string3=[]
|
7 |
for i in range(0,num):
|
8 |
intial="""These are the few examples of converting original sentences into paraphrased sentences.\n original: The gray clouds were a warning of an approaching storm.\n paraphrase: The coming storm was foretold by the dark clouds.\n original: Giraffes like Acacia leaves and hay, and they can consume 75 pounds of food a day.\n paraphrase: A giraffe can eat up to 75 pounds of Acacia leaves and hay daily.\n """
|
9 |
full_input=intial+"original:"+input_sentence + "\n paraphrase:"+start
|
10 |
-
string1=generator(full_input, do_sample=True,max_length= len(full_input.split())+70,min_length=len(full_input.split())+70,temperature=0.65+i*0.05
|
11 |
string2=string1.split('paraphrase:',3)[-1]
|
12 |
string3.append(string2.split('.',1)[0]+".")
|
13 |
return '\n\n'.join([i for i in string3[0:]])
|
|
|
1 |
from transformers import pipeline
|
2 |
import gradio as gr
|
|
|
3 |
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
|
4 |
def query(input_sentence,num,start):
|
5 |
string3=[]
|
6 |
for i in range(0,num):
|
7 |
intial="""These are the few examples of converting original sentences into paraphrased sentences.\n original: The gray clouds were a warning of an approaching storm.\n paraphrase: The coming storm was foretold by the dark clouds.\n original: Giraffes like Acacia leaves and hay, and they can consume 75 pounds of food a day.\n paraphrase: A giraffe can eat up to 75 pounds of Acacia leaves and hay daily.\n """
|
8 |
full_input=intial+"original:"+input_sentence + "\n paraphrase:"+start
|
9 |
+
string1=generator(full_input, do_sample=True,max_length= len(full_input.split())+70,min_length=len(full_input.split())+70,temperature=0.65+i*0.05)[0]['generated_text']
|
10 |
string2=string1.split('paraphrase:',3)[-1]
|
11 |
string3.append(string2.split('.',1)[0]+".")
|
12 |
return '\n\n'.join([i for i in string3[0:]])
|