pp3232133 commited on
Commit
1f6b958
·
1 Parent(s): aa7689a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ token = config.get("token", "")
18
  hf_folder = HfFolder(repo_id, repo_type, token=token)
19
 
20
  # Wczytanie własnego modelu chatbota z Hugging Face
21
- if model_name == "pp3232133/pp3232133-distilgpt2-wikitext2"
22
  tokenizer = AutoTokenizer.from_pretrained(model_name, repo_path=hf_folder)
23
  model = AutoModelForCausalLM.from_pretrained(model_name, repo_path=hf_folder)
24
 
 
18
  hf_folder = HfFolder(repo_id, repo_type, token=token)
19
 
20
  # Wczytanie własnego modelu chatbota z Hugging Face
21
+ if model_name == "pp3232133/pp3232133-distilgpt2-wikitext2":
22
  tokenizer = AutoTokenizer.from_pretrained(model_name, repo_path=hf_folder)
23
  model = AutoModelForCausalLM.from_pretrained(model_name, repo_path=hf_folder)
24