Update app.py

app.py

@@ -15,7 +15,12 @@ try:
     model = PeftModel.from_pretrained(base_model, fine_tuned_model_name)
     
     print("Loading tokenizer...")
-    tokenizer = AutoTokenizer.from_pretrained(base_model_name)
+    try:
+        tokenizer = AutoTokenizer.from_pretrained(base_model_name)
+    except Exception as e:
+        print(f"Failed to load tokenizer for {base_model_name}: {e}")
+        print("Falling back to a generic tokenizer...")
+        tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
 
 except Exception as e:
     print(f"Error loading the model or tokenizer: {e}")