Update app.py

app.py

@@ -6,8 +6,8 @@ import torch
 device = torch.device("cpu")
 
 # Cargar el modelo y tokenizer
-print("Cargando modelo DistilGPT-2...")
-model_name = "distilgpt2"
+print("Cargando modelo code-autocomplete-gpt2-base...")
+model_name = "shibing624/code-autocomplete-gpt2-base"
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)
 
@@ -21,10 +21,10 @@ if tokenizer.pad_token is None:
 
 def autocomplete_text(input_text, max_tokens=20):
     """
-    Autocompleta el texto de entrada usando DistilGPT-2
+    Autocompleta el texto/código de entrada usando code-autocomplete-gpt2-base
     
     Args:
-        input_text (str): Texto inicial a completar
+        input_text (str): Texto/código inicial a completar
         max_tokens (int): Número máximo de tokens a generar
     
     Returns:
@@ -70,29 +70,29 @@ def create_autocomplete_interface():
     Crea la interfaz de autocompletar dentro de gr.Blocks()
     """
     
-    with gr.Blocks(title="Autocompletar Texto") as demo:
+    with gr.Blocks(title="Autocompletar Código") as demo:
         
-        gr.Markdown("# 🤖 Autocompletar Texto")
-        gr.Markdown("Escribe el inicio de una frase y la IA la completará por ti.")
+        gr.Markdown("# 🤖 Autocompletar Código")
+        gr.Markdown("Escribe el inicio de tu código y la IA lo completará por ti.")
         
         with gr.Tab("Autocompletar"):
             with gr.Row():
                 with gr.Column():
                     input_textbox = gr.Textbox(
-                        label="Texto a completar",
-                        placeholder="Escribe el inicio de tu frase aquí...",
-                        lines=3,
-                        max_lines=5
+                        label="Código a completar",
+                        placeholder="def fibonacci(n):",
+                        lines=5,
+                        max_lines=10
                     )
                     
-                    generate_btn = gr.Button("Completar Texto", variant="primary")
+                    generate_btn = gr.Button("Completar Código", variant="primary")
                     
                 with gr.Column():
                     output_textbox = gr.Textbox(
-                        label="Texto generado",
-                        placeholder="Aquí aparecerá la continuación...",
-                        lines=3,
-                        max_lines=5,
+                        label="Código generado",
+                        placeholder="Aquí aparecerá la continuación del código...",
+                        lines=5,
+                        max_lines=10,
                         interactive=False
                     )
             
@@ -115,21 +115,24 @@ def create_autocomplete_interface():
             gr.Markdown("""
             ### Ejemplos de uso:
             
-            **Entrada:** "El clima de hoy está"
-            **Salida:** "muy agradable y soleado"
+            **Entrada:** "def fibonacci(n):"
+            **Salida:** "\\n    if n <= 1:\\n        return n\\n    return fibonacci(n-1) + fibonacci(n-2)"
             
-            **Entrada:** "Me gusta mucho"
-            **Salida:** "pasar tiempo con mi familia"
+            **Entrada:** "for i in range("
+            **Salida:** "10):\\n    print(i)"
             
-            **Entrada:** "Para hacer una buena comida necesitas"
-            **Salida:** "ingredientes frescos y mucha paciencia"
+            **Entrada:** "import pandas as pd\\ndf = pd.read_csv("
+            **Salida:** "'data.csv')\\nprint(df.head())"
+            
+            **Entrada:** "class Calculator:"
+            **Salida:** "\\n    def __init__(self):\\n        pass"
             """)
     
     return demo
 
 # Crear y lanzar la aplicación
 if __name__ == "__main__":
-    print("Iniciando aplicación de autocompletar...")
+    print("Iniciando aplicación de autocompletar código...")
     
     # Crear la interfaz
     app = create_autocomplete_interface()