Spaces:
Running
Running
File size: 2,288 Bytes
a4287ea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
#!/usr/bin/env python3
"""
Script de teste para verificar se todas as dependências estão funcionando
"""
def test_imports():
"""Testa se todas as importações funcionam"""
try:
import gradio as gr
print("✓ Gradio importado com sucesso")
print(f" Versão do Gradio: {gr.__version__}")
except ImportError as e:
print(f"✗ Erro ao importar Gradio: {e}")
return False
try:
from transformers import TrOCRProcessor, VisionEncoderDecoderModel
print("✓ Transformers importado com sucesso")
except ImportError as e:
print(f"✗ Erro ao importar Transformers: {e}")
return False
try:
from PIL import Image
print("✓ PIL/Pillow importado com sucesso")
except ImportError as e:
print(f"✗ Erro ao importar PIL: {e}")
return False
try:
import torch
print("✓ PyTorch importado com sucesso")
print(f" Versão do PyTorch: {torch.__version__}")
except ImportError as e:
print(f"✗ Erro ao importar PyTorch: {e}")
return False
return True
def test_model_loading():
"""Testa se o modelo pode ser carregado"""
try:
from transformers import TrOCRProcessor, VisionEncoderDecoderModel
print("Testando carregamento do modelo...")
processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
print("✓ Modelo carregado com sucesso")
return True
except Exception as e:
print(f"✗ Erro ao carregar modelo: {e}")
return False
if __name__ == "__main__":
print("🧪 Executando testes do Professor IA")
print("=" * 50)
if test_imports():
print("\n📦 Todas as importações estão funcionando!")
if test_model_loading():
print("\n🤖 Modelo carregado com sucesso!")
print("\n✅ Todos os testes passaram! O aplicativo deve funcionar no Hugging Face.")
else:
print("\n⚠️ Problema com o carregamento do modelo.")
else:
print("\n❌ Problemas com as importações. Verifique o requirements.txt") |