dar-tau commited on
Commit
8f35fab
1 Parent(s): ce47361

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -188,7 +188,7 @@ def run_interpretation(raw_original_prompt, raw_interpretation_prompt, max_new_t
188
 
189
  ## main
190
  torch.set_grad_enabled(False)
191
- model_name = 'GPT-J'
192
  raw_original_prompt = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
193
  tokens_container = []
194
 
 
188
 
189
  ## main
190
  torch.set_grad_enabled(False)
191
+ model_name = 'LLAMA2-13B'
192
  raw_original_prompt = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
193
  tokens_container = []
194