Keyurjotaniya007 commited on
Commit
bdbc895
·
verified ·
1 Parent(s): 5b03cbf

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +45 -0
  2. requirements.txt +5 -0
app.py ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline, set_seed
2
+
3
+ import gradio as gr
4
+
5
+ # Carica il modello GPT-2
6
+
7
+ generator = pipeline('text-generation', model='gpt2')
8
+
9
+ set_seed(42)
10
+
11
+ def generate_text(prompt):
12
+
13
+ response = generator(
14
+
15
+ prompt,
16
+
17
+ max_length=100,
18
+
19
+ num_return_sequences=1,
20
+
21
+ truncation=True
22
+
23
+ )
24
+
25
+ return response[0]['generated_text']
26
+
27
+ # Interfaccia Gradio
28
+
29
+ interface = gr.Interface(
30
+
31
+ fn=generate_text,
32
+
33
+ inputs=gr.Textbox(lines=5, placeholder="Inserisci il tuo testo iniziale..."),
34
+
35
+ outputs="text",
36
+
37
+ title="GPT-2 Text Generator",
38
+
39
+ description="Genera testo continuativo usando GPT-2 senza fine-tuning."
40
+
41
+ )
42
+
43
+ if __name__ == "__main__":
44
+
45
+ interface.launch()
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ transformers
2
+
3
+ gradio
4
+
5
+ torch