Update app.py

app.py

@@ -15,9 +15,9 @@ def load_model():
     try:
         print("Loading model from safetensors format...")
         
-        # Load tokenizer and model from local directory
-        tokenizer = AutoTokenizer.from_pretrained("./summarization_model")
-        model = AutoModelForSeq2SeqLM.from_pretrained("./summarization_model")
+        # Load tokenizer and model from current directory (root of Space)
+        tokenizer = AutoTokenizer.from_pretrained(".")
+        model = AutoModelForSeq2SeqLM.from_pretrained(".")
         
         print("✓ Model and tokenizer loaded successfully")
         print(f"  Model parameters: {model.num_parameters():,}")
@@ -35,7 +35,8 @@ def load_model():
     
     except Exception as e:
         print(f"❌ Error loading model: {e}")
-        print("Make sure the model files (model.safetensors, config.json, tokenizer.json) are in the ./summarization_model directory")
+        print("Make sure the model files (model.safetensors, config.json, tokenizer.json) are in the root directory")
+        print(f"Current directory contents: {os.listdir('.')}")
         return None
 
 # Load model at startup
@@ -57,7 +58,7 @@ def summarize_text(
     Summarize the input text with customizable parameters
     """
     if summarizer is None:
-        return "❌ Error: Model not loaded. Please check model files in ./summarization_model directory."
+        return "❌ Error: Model not loaded. Please check model files (model.safetensors, config.json, tokenizer.json) in the root directory."
     
     if not text.strip():
         return "⚠️ Please enter some text to summarize."