File size: 1,202 Bytes
22ca508 |
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 |
#!/usr/bin/env python3
"""
Script de prueba para verificar que el GPT Local funciona
"""
import sys
from pathlib import Path
# Agregar el directorio raíz al path
sys.path.append(str(Path(__file__).parent))
try:
from models.model_loader import ModelLoader
from models.text_generator import TextGenerator
print("🚀 Probando GPT Local...")
# Inicializar componentes
model_loader = ModelLoader()
text_generator = TextGenerator(model_loader)
# Cargar modelo pequeño
print("📦 Cargando modelo DialoGPT-small...")
success = model_loader.load_model("microsoft/DialoGPT-small")
if success:
print("✅ Modelo cargado exitosamente")
# Prueba básica
test_message = "Hola"
print(f"🧪 Probando con: '{test_message}'")
response = text_generator.generate_response(test_message, max_length=50)
print(f"🤖 Respuesta: {response}")
print("\n✅ ¡GPT Local funcionando correctamente!")
else:
print("❌ Error al cargar el modelo")
except Exception as e:
print(f"❌ Error: {str(e)}")
import traceback
traceback.print_exc()
|