test_deps.py

#!/usr/bin/env python3
"""
Test script to verify the transformers dependencies are working
"""

def test_imports():
    """Test that all required transformers imports work"""
    try:
        print("Testing transformers imports...")
        
        from transformers import AutoProcessor, Gemma3nForConditionalGeneration
        print("✅ Gemma3nForConditionalGeneration import successful")
        
        from transformers import AutoTokenizer, AutoModelForCausalLM
        print("✅ Standard transformers imports successful")
        
        import torch
        print("✅ PyTorch import successful")
        
        import sentencepiece
        print("✅ SentencePiece import successful")
        
        import tiktoken
        print("✅ TikToken import successful")
        
        import protobuf
        print("✅ Protobuf import successful")
        
        print("\n🎉 All imports successful! Ready for Hugging Face Spaces deployment")
        return True
        
    except ImportError as e:
        print(f"❌ Import error: {e}")
        return False

if __name__ == "__main__":
    test_imports()