Spaces:

wakeupmh
/

ama-autism

Sleeping

wakeupmh commited on Feb 16

Commit

1436681

1 Parent(s): 4ee9d54

refactor: try deepseek

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ logging.basicConfig(level=logging.INFO)
 DATA_DIR = "/data" if os.path.exists("/data") else "."
 DATASET_DIR = os.path.join(DATA_DIR, "rag_dataset")
 DATASET_PATH = os.path.join(DATASET_DIR, "dataset")
-MODEL_PATH = "hysts/mistral-7b"
 @st.cache_resource
 def load_local_model():

 DATA_DIR = "/data" if os.path.exists("/data") else "."
 DATASET_DIR = os.path.join(DATA_DIR, "rag_dataset")
 DATASET_PATH = os.path.join(DATASET_DIR, "dataset")
+MODEL_PATH = "deepseek-ai/DeepSeek-R1-Distill-Llama-8B"  # Using flan-t5-small for better performance
 @st.cache_resource
 def load_local_model():