Update app.py

app.py

@@ -3,19 +3,15 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from joblib import Memory
 import datetime
-import os
-
-# Enable offline mode for Hugging Face transformers
-os.environ["HF_HUB_OFFLINE"] = "1"
 
 # Initialize cache
 cache_dir = "./cache"
 memory = Memory(cache_dir, verbose=0)
 
-# Load pre-trained model and tokenizer with local_files_only=True
+# Load pre-trained model and tokenizer (allow online download)
 model_name = "distilgpt2"
-tokenizer = AutoTokenizer.from_pretrained(model_name, local_files_only=True)
-model = AutoModelForCausalLM.from_pretrained(model_name, local_files_only=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Set pad_token_id to eos_token_id to avoid warnings
 tokenizer.pad_token = tokenizer.eos_token