Update app.py

app.py

@@ -210,9 +210,7 @@ async def generate_model_response(model, inputs, max_tokens=2048):
         text = remove_duplicates(response['choices'][0]['text'])
         
         if len(text.split()) > max_tokens:
-            parts = [text[i:i + max_tokens] for i in range(0, len(text), max_tokens)]
-            return ' '.join(parts)
-        
+            return text[:max_tokens]
         return text
     except Exception as e:
         return ""
@@ -231,13 +229,10 @@ async def process_message(message):
     best_response = get_best_response(responses)
     return best_response
 
-class MessageRequest(BaseModel):
-    message: str
-
 app = FastAPI()
 
 @app.post("/generate")
-async def generate(request: MessageRequest):
+async def generate(request: BaseModel):
     try:
         response = await process_message(request.message)
         return JSONResponse(content={"response": response})