Update app.py

app.py

@@ -12,7 +12,7 @@ gpu = 0 if torch.cuda.is_available() else -1
 # Initialize the text generation pipeline
 # This function will be able to generate text
 # given an input.
-pipe = pipeline("text2text-generation", model="google/flan-t5-large",trust_remote_code=True,device=gpu)
+pipe = pipeline("text2text-generation", model="gpt2",trust_remote_code=True,device=gpu)
 
 # Define a function to handle the GET request at `/generate`
 # The generate() function is defined as a FastAPI route that takes a