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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -17,9 +17,9 @@ token = config["token", ""]
17
  hf_folder = HfFolder(repo_id, repo_type, token=token)
18
 
19
  # Wczytanie własnego modelu chatbota z Hugging Face
20
- 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
 
24
  # Funkcja obsługująca wejście i wyjście dla interfejsu Gradio
25
  def chatbot_interface(input_text):
 
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
 
24
  # Funkcja obsługująca wejście i wyjście dla interfejsu Gradio
25
  def chatbot_interface(input_text):