FIX: Add proper test_fixed.py with from_pretrained support

test_fixed.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python3
+"""
+Test script for the FIXED pixeltext-ai model
+"""
+
+from transformers import AutoModel
+from PIL import Image, ImageDraw, ImageFont
+
+def test_fixed_model():
+    """Test the fixed model."""
+    
+    print("🧪 Testing FIXED pixeltext-ai model...")
+    
+    # Load model from Hub
+    model = AutoModel.from_pretrained("BabaK07/pixeltext-ai", trust_remote_code=True)
+    
+    # Create test image
+    img = Image.new('RGB', (400, 200), color='white')
+    draw = ImageDraw.Draw(img)
+    
+    try:
+        font = ImageFont.truetype("/System/Library/Fonts/Arial.ttf", 20)
+    except:
+        font = ImageFont.load_default()
+    
+    draw.text((20, 50), "FIXED MODEL TEST", fill='black', font=font)
+    draw.text((20, 100), "Hub loading works!", fill='blue', font=font)
+    draw.text((20, 150), "from_pretrained success!", fill='green', font=font)
+    
+    # Extract text
+    result = model.generate_ocr_text(img)
+    
+    print(f"📝 Results:")
+    print(f"   Text: {result['text']}")
+    print(f"   Confidence: {result['confidence']:.1%}")
+    print(f"   Success: {result['success']}")
+    print(f"   Method: {result['method']}")
+    
+    if result['success']:
+        print("✅ FIXED MODEL WORKING PERFECTLY!")
+    else:
+        print("❌ Still has issues")
+    
+    return result
+
+if __name__ == "__main__":
+    test_fixed_model()