dar-tau commited on
Commit
bf5b0c8
·
verified ·
1 Parent(s): 3e51fa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -186,8 +186,7 @@ def run_interpretation(raw_original_prompt, raw_interpretation_prompt, max_new_t
186
 
187
  ## main
188
  torch.set_grad_enabled(False)
189
- global_state = GlobalState()
190
-
191
  model_name = 'LLAMA2-7B'
192
  reset_model(model_name, load_on_gpu=True, with_extra_components=False, reset_sentence_transformer=True)
193
  raw_original_prompt = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
 
186
 
187
  ## main
188
  torch.set_grad_enabled(False)
189
+ global_state = gr.State(GlobalState)
 
190
  model_name = 'LLAMA2-7B'
191
  reset_model(model_name, load_on_gpu=True, with_extra_components=False, reset_sentence_transformer=True)
192
  raw_original_prompt = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')