Momo47-m commited on
Commit
126589e
·
verified ·
1 Parent(s): 9326c38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -5,9 +5,8 @@ from transformers import pipeline
5
  generator = pipeline("text-generation", model="distilgpt2", device=0) # Usa la GPU se disponibile (device=0)
6
 
7
  def generate_c_code(prompt):
8
- # Aggiungi un contesto specifico per la generazione del codice C
9
- prompt = f"Scrivi un programma in C che {prompt}"
10
- # Limita la lunghezza del testo generato per migliorare la velocità
11
  generated = generator(prompt, max_length=100, num_return_sequences=1)
12
  return generated[0]["generated_text"]
13
 
@@ -19,4 +18,4 @@ iface = gr.Interface(fn=generate_c_code,
19
  description="Chiedi tutto ciò che vuoi e Bakloul te lo crea in C:")
20
 
21
  # Esegui l'interfaccia
22
- iface.launch()
 
5
  generator = pipeline("text-generation", model="distilgpt2", device=0) # Usa la GPU se disponibile (device=0)
6
 
7
  def generate_c_code(prompt):
8
+ # Specifica chiaramente il tipo di programma C che vogliamo generare
9
+ prompt = f"Scrivi un programma in C che {prompt}" # Prompt più chiaro per generare codice C
 
10
  generated = generator(prompt, max_length=100, num_return_sequences=1)
11
  return generated[0]["generated_text"]
12
 
 
18
  description="Chiedi tutto ciò che vuoi e Bakloul te lo crea in C:")
19
 
20
  # Esegui l'interfaccia
21
+ iface.launch()