ahmadmac commited on
Commit
307cc82
·
verified ·
1 Parent(s): e2683ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,6 +4,7 @@ import gradio as gr
4
  tokenizer = AutoTokenizer.from_pretrained("ahmadmac/Pretrained-GPT2")
5
  model = AutoModelForCausalLM.from_pretrained("ahmadmac/Pretrained-GPT2")
6
 
 
7
  def generate_text(prompt):
8
  inputs = tokenizer(prompt, return_tensors="pt")
9
  outputs = model.generate(**inputs, max_length=50)
@@ -17,4 +18,5 @@ iface = gr.Interface(
17
  title="GPT-2 Text Generator",
18
  description="Enter a prompt to generate text using GPT-2"
19
  )
20
- iface.launch()
 
 
4
  tokenizer = AutoTokenizer.from_pretrained("ahmadmac/Pretrained-GPT2")
5
  model = AutoModelForCausalLM.from_pretrained("ahmadmac/Pretrained-GPT2")
6
 
7
+
8
  def generate_text(prompt):
9
  inputs = tokenizer(prompt, return_tensors="pt")
10
  outputs = model.generate(**inputs, max_length=50)
 
18
  title="GPT-2 Text Generator",
19
  description="Enter a prompt to generate text using GPT-2"
20
  )
21
+
22
+ iface.launch(server_port=7861)