Update app.py

app.py

@@ -11,9 +11,9 @@ app = FastAPI()
 def greet_json():
     return {"Hello": "World!"}
 
-@app.get("/message?input={0}")
-def message(req):
-    inputs = tokenizer(req.query.input, return_tensors="pt", padding=True, truncation=True)
+@app.get("/message?input={prompt}")
+def message(prompt: str):
+    inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
     output = model.generate(
         input_ids=inputs["input_ids"],
         attention_mask=inputs["attention_mask"],