Update app.py

app.py

@@ -7,11 +7,13 @@ from transformers import pipeline
 app = FastAPI(docs_url="/")
 
 @app.get("/generate")
-def generate(text: str):
+def generate(breed: str):
     """
     Using the text2text-generation pipeline from `transformers`, generate text
     from the given input text. The model used is `google/flan-t5-small`, which
     can be found [here](https://huggingface.co/google/flan-t5-small).
     """
-    
-    return "Dog is healthy!"
+    if(breed=='Labro')
+    return "Dog is healthy!"
+    else
+    return "Dog is super healthy!"