Update app.py

app.py

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