simple modelo

app.py

@@ -1,8 +1,10 @@
 from transformers import pipeline
 import gradio as gr
+import torch
 
 # Cargar un modelo de texto de Hugging Face (puede ser de completado, traducción, etc.)
-modelo = pipeline("text-generation", model="microsoft/Phi-4-multimodal-instruct")  # Cambia el modelo si lo deseas
+dispositivo = 0 if torch.cuda.is_available() else -1
+modelo = pipeline("text-generation", model="microsoft/Phi-4-multimodal-instruct", device=dispositivo)  # Cambia el modelo si lo deseas
 
 # Función para generar texto
 def generar_texto(prompt):
@@ -20,4 +22,4 @@ interfaz = gr.Interface(
 
 # Ejecutar la app
 if __name__ == "__main__":
-    interfaz.launch()
+    interfaz.launch()