Update app.py

app.py

@@ -41,23 +41,21 @@ def sample_from(logits):
   preds = np.asarray(preds).astype("float32")
   return np.random.choice(i, p=preds)
 
-def generate_answers(start_prompt):
+def generate_answers(text):
   num_tokens_generated = 0 
-  sample_index = len(start_prompt) - 1
-  start_tokens = [word_to_index.get(_, 1) for _ in start_prompt]
+  sample_index = len([text]) - 1
   tokens_generated= []
 
-  text_out = text_process_pipeline(start_prompt)
+  text_out = text_process_pipeline([text])
   predictions,_ = model.predict(text_out)
   results = np.argmax(predictions, axis=1)[0]
 
   while num_tokens_generated <= 40:
     sample_token = sample_from(predictions[0][sample_index])
     tokens_generated.append(sample_token)
-    start_tokens.append(sample_token)
     num_tokens_generated = len(tokens_generated)
 
-  text_out = tokenizer.sequences_to_texts([tokens_generated])  
+  text_out = tokenizer.sequences_to_texts([tokens_generated])
   return text_out[0]
     
 examples = [["The movie was nice, "], ["It was showing nothing special to "]]