Update app.py

app.py

@@ -13,7 +13,8 @@ def query(input_sentence,num,start):
        data = json.dumps({"inputs":full_input,"parameters":{"max_length":len(full_input.split())+70,"min_length":len(full_input.split())+70},"temperature":0.65+0.05*i})
        response = requests.request("POST", API_URL, headers=headers, data=data)
        output=json.loads(response.content.decode("utf-8"))[0]['generated_text']
-       paraphrase_text=output.split('paraphrase:',3)[-1]
+       paraphrase=output.split('paraphrase:',3)[-1]
+       paraphrase_text=paraphrase.split('original:',1)[0]
        paraphrase_final.append( paraphrase_text.split('.',1)[0]+".")
     return '\n\n'.join([i for i in paraphrase_final[0:]])
 title = "Paraphrasing with GPT-NEO"
@@ -28,5 +29,4 @@ title=title,description=description,
 article= article,
 examples=examples,
 allow_screenshot=False, 
-allow_flagging=False,
-enable_queue=True).launch()   
+allow_flagging=False).launch(enable_queue=True)