Update app.py

app.py

@@ -135,11 +135,11 @@ def chitchatRu():
     try:
         text = request.form.get('text') or request.args.get('text') or request.values.get('text') or ""
         if text == "": return {"status": "error", "details": { "error_code": 101, "error_details": "No text provided" }}
-
+        
         inputs = chct_t(text, padding=True, truncation=True, return_tensors="pt")
-        outputs = chct_m(**inputs)
-        predicted_class = outputs.logits.argmax(dim=1).item()
-        answer = chct_t.decode(predicted_class)
+        
+        generated_ids = chct_m.generate(input_ids=input_ids)
+        answer = chct_t.decode(generated_ids[0], skip_special_tokens=True)
     
         return {"status": "pass", "answer": answer}
     except Exception as e: return {"status": "error", "details": { "error_code": 123, "error_details": str(e).replace("\n", " | ") }}