Update app.py
Browse files
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')
|