fix: local model

services/model_handler.py

@@ -4,15 +4,13 @@ import streamlit as st
 from agno.agent import Agent
 from agno.tools.arxiv import ArxivTools
 from agno.tools.pubmed import PubmedTools
-from agno.models.base import BaseModel
-import os
-import time
+from agno.models.base import Model
 from tenacity import retry, stop_after_attempt, wait_exponential
 
 MODEL_PATH = "google/flan-t5-small"
 
 # Personnalized class for local models
-class LocalHuggingFaceModel(BaseModel):
+class LocalHuggingFaceModel(Model):
     def __init__(self, model, tokenizer, max_length=512):
         super().__init__()
         self.model = model