Update app.py

app.py

@@ -9,22 +9,17 @@ with open("config.txt", "r") as f:
         key, value = line.strip().split(" = ")
         config[key] = value
 
-repo_id = config["repo_id"]
-repo_type = config["repo_type"]
-token = config["token"]
+repo_id = config["repo_id", ""]
+repo_type = config["repo_type", ""]
+token = config["token", ""]
 
 # Utworzenie folderu Hugging Face z tokenem uwierzytelniającym
 hf_folder = HfFolder(repo_id, repo_type, token=token)
 
-# Wczytanie modelu chatbota z Hugging Face
-model_name = "pp3232133/distilgpt2-wikitext2"
-tokenizer = AutoTokenizer.from_pretrained(model_name, repo_path=hf_folder)
-model = AutoModelForCausalLM.from_pretrained(model_name, repo_path=hf_folder)
-
 # Wczytanie własnego modelu chatbota z Hugging Face
 model_name = "pp3232133/pp3232133-distilgpt2-wikitext2"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, repo_path=hf_folder)
+model = AutoModelForCausalLM.from_pretrained(model_name, repo_path=hf_folder)
 
 # Funkcja obsługująca wejście i wyjście dla interfejsu Gradio
 def chatbot_interface(input_text):
@@ -45,3 +40,7 @@ iface = gr.Interface(
 
 # Uruchomienie interfejsu
 iface.launch()
+
+
+else:
+    print("Nie można znaleźć nazwy modelu w pliku konfiguracyjnym.")