Hugging Face's logo

Spaces:
Percy3822
/
model_trainer
Sleeping

App Files Community
Percy3822 commited on
Commit
5bab387
·
verified ·
1 Parent(s): d3870d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import zipfile
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
  import torch
8
 
9
- model_dir = Path(_file_).parent / "trained_model"
10
 
11
  # Gracefully fail if model not trained yet
12
  if not model_dir.exists():
 
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
  import torch
8
 
9
+ model_dir = Path(__file__).parent / "trained_model"
10
 
11
  # Gracefully fail if model not trained yet
12
  if not model_dir.exists():