Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
-pipe = pipeline("text-generation", model="poem_generator2", device=device)
+pipe = pipeline("text-generation", model="poem_generator3", device=device)
 
 def generate_poem(text):
     res = pipe(text)[0]["generated_text"]