THEODOROS commited on
Commit
d27b2fd
·
1 Parent(s): 1084c30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -12,8 +12,7 @@ import gradio as gr
12
 
13
  from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM
14
 
15
- #finetuned = AutoModelForCausalLM.from_pretrained('model')
16
- finetuned = AutoModelForCausalLM.from_pretrained('.')
17
  tokenizer = AutoTokenizer.from_pretrained('gpt2')
18
 
19
  # Utility functions
@@ -251,7 +250,7 @@ iface = gr.Interface(fn=gen_and_mutate, inputs=[textbox, checkbox, topp_slider,
251
  <li> <strong>location</strong>: "a house with a bedroom in the north east side"</li>
252
  </ul>
253
  <p>You can also create a mutation of the generated layout by enabling the 'Mutate' option.</p>
254
- <p> Made by: <a href='https://www.linkedin.com/in/theodorosgalanos/'>Theodoros </a> <a href='https://twitter.com/TheodoreGalanos'> Galanos</a> and <a href='https://twitter.com/tylerlastovich'>Tyler Lastovich</a> </p>
255
  </div>''')
256
 
257
  iface.launch()
 
12
 
13
  from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM
14
 
15
+ finetuned = AutoModelForCausalLM.from_pretrained('model')
 
16
  tokenizer = AutoTokenizer.from_pretrained('gpt2')
17
 
18
  # Utility functions
 
250
  <li> <strong>location</strong>: "a house with a bedroom in the north east side"</li>
251
  </ul>
252
  <p>You can also create a mutation of the generated layout by enabling the 'Mutate' option.</p>
253
+ <p> Made by: <a href='https://www.linkedin.com/in/theodorosgalanos/'>Theodoros </a> <a href='https://twitter.com/TheodoreGalanos'> Galanos</a> and <a href='https://twitter.com/tylerlastovich'>Tyler Lastovich</a>, using a finetuned <a href='https://huggingface.co/EleutherAI/gpt-neo-125M'> GPT-Neo</a> model. </p>
254
  </div>''')
255
 
256
  iface.launch()