Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 
-# Usa un modello più leggero e veloce, come distilgpt2
-generator = pipeline('text-generation', model='distilgpt2')
+# Usa CodeT5, che è specializzato per il codice
+generator = pipeline('text2text-generation', model='Salesforce/codeT5-large')
 
 # Funzione per generare il codice C
 def genera_codice(input_text):
-    # Prompt migliorato per guidare meglio il modello
+    # Prompt migliorato per generare codice C
     prompt = f"Scrivi un programma in C che risolva il problema seguente: {input_text}"
     result = generator(prompt, max_length=100, num_return_sequences=1)
     codice_generato = result[0]['generated_text']