Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, HfFolder
 
 # Wczytanie tokena z pliku konfiguracyjnego
 with open("config.txt", "r") as f:
@@ -15,7 +15,7 @@ repo_type = config.get("repo_type", "")
 token = config.get("token", "")
 
 # Utworzenie folderu Hugging Face z tokenem uwierzytelniającym
-hf_folder = HfFolder(repo_id, repo_type, token=token)
+hf_folder = HfFolder(model_name, repo_type, token=token)
 
 # Wczytanie własnego modelu chatbota z Hugging Face
 if model_name == "pp3232133/pp3232133-distilgpt2-wikitext2":
@@ -37,11 +37,10 @@ if model_name == "pp3232133/pp3232133-distilgpt2-wikitext2":
         title="Chatbot",
         description="Custom chatbot based on your Hugging Face model. Start typing to chat with the bot.",
         theme="compact"
-        )
+    )
 
     # Uruchomienie interfejsu
     iface.launch()
 
-
 else:
-    print("Nie można znaleźć nazwy modelu w pliku konfiguracyjnym.")
+    print("Nie można znaleźć nazwy modelu w pliku konfiguracyjnym.")