hardik8588 commited on
Commit
9fa6273
·
verified ·
1 Parent(s): 20ce10a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -104,8 +104,8 @@ def load_document_context(task_id):
104
 
105
 
106
  load_dotenv()
107
- DB_PATH = os.getenv("DB_PATH", os.path.join(os.path.dirname(__file__), "data/user_data.db"))
108
- os.makedirs(os.path.join(os.path.dirname(__file__), "data"), exist_ok=True)
109
 
110
  def fine_tune_qa_model():
111
  """Fine-tunes a QA model on the CUAD dataset."""
 
104
 
105
 
106
  load_dotenv()
107
+ DB_PATH = os.getenv("DB_PATH", os.path.abspath("/tmp/data/user_data.db"))
108
+ os.makedirs(os.path.dirname(DB_PATH), exist_ok=True)
109
 
110
  def fine_tune_qa_model():
111
  """Fine-tunes a QA model on the CUAD dataset."""