rabiyulfahim commited on
Commit
820d571
·
verified ·
1 Parent(s): 484d307

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ os.environ["HF_HOME"] = "/tmp"
13
  os.environ["TRANSFORMERS_CACHE"] = "/tmp"
14
 
15
 
16
- model_id = "rabiyulfahim/qa_python_gpt2"
17
 
18
  tokenizer = AutoTokenizer.from_pretrained(model_id, cache_dir="/tmp")
19
  model = AutoModelForCausalLM.from_pretrained(model_id, cache_dir="/tmp")
 
13
  os.environ["TRANSFORMERS_CACHE"] = "/tmp"
14
 
15
 
16
+ model_id = "rabiyulfahim/Python_coding_model"
17
 
18
  tokenizer = AutoTokenizer.from_pretrained(model_id, cache_dir="/tmp")
19
  model = AutoModelForCausalLM.from_pretrained(model_id, cache_dir="/tmp")