pp3232133 commited on
Commit
b6384c5
·
1 Parent(s): 09b8877

Update app.py

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