Update app.py

app.py

@@ -75,7 +75,7 @@ def initialize_gpu():
     pass
 
 
-def reset_model(model_name, demo_blocks=None): 
+def reset_model(model_name, demo_blocks): 
     # extract model info
     model_args = deepcopy(model_info[model_name])
     model_path = model_args.pop('model_path')
@@ -148,7 +148,7 @@ torch.set_grad_enabled(False)
 global_state = GlobalState()
 
 model_name = 'LLAMA2-7B'
-reset_model(model_name)
+reset_model(model_name, None)
 original_prompt_raw = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
 tokens_container = []