Update app.py

app.py

@@ -7,10 +7,12 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer, pipeline
 
 os.environ["TOKENIZERS_PARALLELISM"] = "true"
 generator = pipeline('text-generation', model="salomonsky/deepSP")
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 
 def generate_output(text):
     prompt = ""
-    input_text = prompt + ""
+    input_tokens = tokenizer.encode(text, add_special_tokens=False)
+    input_text = tokenizer.decode(input_tokens)
     gpt2_output = generator(input_text, max_length=20, do_sample=True, temperature=0.9)
     generated_text = gpt2_output[0]['generated_text']
     generated_text = generated_text.replace(input_text, "").strip()