Spaces:

wakeupmh
/

ama-autism

Sleeping

wakeupmh commited on Feb 16

Commit

4ee9d54

1 Parent(s): f90b4a9

refactor: try mistral

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 = "google/flan-t5-small"  # Using flan-t5-small for better performance
 @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 = "hysts/mistral-7b"
 @st.cache_resource
 def load_local_model():