Update app.py

app.py

@@ -1,8 +1,8 @@
 from fastapi import FastAPI
-# from transformers import AutoModel, AutoTokenizer
+from transformers import AutoModel, AutoTokenizer
 
-# tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-0.6B")
-# model = AutoModel.from_pretrained("Qwen/Qwen3-0.6B")
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-0.6B")
+model = AutoModel.from_pretrained("Qwen/Qwen3-0.6B")
 
 app = FastAPI()
 
@@ -12,4 +12,10 @@ def greet_json():
 
 @app.get("/message?input={0}")
 def message(req):
-    return req.query.input
+    inputs = tokenizer(req.query.input, return_tensors="pt", padding=True, truncation=True)
+    output = model.generate(
+        input_ids=inputs["input_ids"],
+        attention_mask=inputs["attention_mask"],
+        temperature=0.3
+    )
+    return tokenizer.decode(output[0], skip_special_tokens=True)