Update main.py

main.py

@@ -1,7 +1,6 @@
 from dotenv import load_dotenv
 import os
 import json
-import requests
 import redis
 from transformers import (
     AutoTokenizer,
@@ -186,12 +185,6 @@ async def process(request: Request):
     else:
         raise HTTPException(status_code=400, detail="Request must contain 'train' or 'message'.")
 
-def get_chatbot_response(user_id, question, predicted_class, language):
-    if user_id not in conversation_history:
-        conversation_history[user_id] = []
-    conversation_history[user_id].append(question)
-    return chatbot_service.get_response(user_id, question, language)
-
 @app.get("/")
 async def get_home():
     user_id = str(uuid.uuid4())
@@ -357,9 +350,11 @@ def train_unified_model():
                         loss.backward()
                         optimizer.step()
 
-                model_data_bytes = torch.save(model.state_dict(), "model_data.pt")
+                model_data_path = "model_data.pt"
+                torch.save(model.state_dict(), model_data_path)
+                with open(model_data_path, "rb") as f:
+                    model_data_bytes = f.read()
                 redis_client.set(f"model:unified_model", model_data_bytes)
-
                 redis_client.delete("training_queue")
         time.sleep(60)
 
@@ -367,4 +362,4 @@ if __name__ == "__main__":
     training_process = multiprocessing.Process(target=train_unified_model)
     training_process.start()
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=8000)