Update app.py

app.py

@@ -8,12 +8,18 @@ model = GPT2LMHeadModel.from_pretrained('mindwrapped/gpt2-lotr-fellowship')
 
 
 def generate_text(text, temperature):
-  generated = tokenizer("<|startoftext|> " + text, return_tensors="pt").input_ids
-  sample_outputs = model.generate(generated, do_sample=True, top_k=50, 
-                                  max_length=300, top_p=0.95, temperature=float(temperature), num_return_sequences=1)
-
-  for i, sample_output in enumerate(sample_outputs):
-    out = tokenizer.decode(sample_output, skip_special_tokens=True)
+  stop = False
+  out = text
+  while not stop:
+    generated = tokenizer("<|startoftext|> " + out, return_tensors="pt").input_ids
+    sample_outputs = model.generate(generated, do_sample=True, top_k=50, 
+                                    max_length=300, top_p=0.95, temperature=float(temperature), num_return_sequences=1)
+  
+    for i, sample_output in enumerate(sample_outputs):
+      out = tokenizer.decode(sample_output, skip_special_tokens=True)
+      
+    if len(out) > 50:
+      stop = True
 
   return out