Update app.py

app.py

@@ -41,7 +41,7 @@ def model_generate(text):
     # generate some output based on the context
     #context = torch.tensor(np.array(encode("Hello! My name is ", tokenizer)))
     #context = torch.zeros((1, 1), dtype=torch.long, device=DEVICE)
-    text_input = str(input())
+    text_input = str(text)
     context_np = np.array(encode(text_input, tokenizer))
     context_np = np.array([context_np])
     context = torch.from_numpy(context_np)