Update app.py

app.py

@@ -3,16 +3,15 @@ import torch
 from transformers import pipeline
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!"
+def greet(txt):
+    device = torch.device("cpu")
+    pipe = pipeline(
+    "text-generation", model="checkpoint-5000", device=device
+)
+    return txt + " "+ pipe(txt, num_return_sequences=1)[0]["generated_text"]
 
 demo = gr.Interface(fn=greet, inputs="text", outputs="text")
     
 demo.launch()  
 # device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-device = torch.device("cpu")
-pipe = pipeline(
-    "text-generation", model="/content/distilgpt2-finetuned-causal/checkpoint-5000", device=device
-)
-txt = "I do not like going to bank but I like"
-print(pipe(txt, num_return_sequences=1)[0]["generated_text"])
+