Change model to Question2WrongAnswer

app.py

@@ -8,7 +8,11 @@ app = FastAPI()
 # This function will be able to generate text
 # given an input.
 pipe = pipeline("text2text-generation", 
-model="google/flan-t5-small")
+model="hirokibastos/Question2WrongAnswer")
+
+@app.get("/")
+def read_root():
+    return {"Hello": "World!"}
  
 # Define a function to handle the GET request at `/generate`
 # The generate() function is defined as a FastAPI route that takes a 
@@ -25,4 +29,5 @@ def generate(text: str):
     output = pipe(text)
      
     # Return the generated text in a JSON response
-    return {"output": output[0]["generated_text"]}
+    return {"output": output[0]["generated_text"]}
+