Update app.py

app.py

@@ -57,7 +57,7 @@ def reset_model(model_name, *extra_components, with_extra_components=True):
     # get model
     global_state.model, global_state.tokenizer, global_state.hidden_states = None, None, None
     gc.collect()
-    global_state.model = AutoModelClass.from_pretrained(model_path, **model_args).cuda()
+    global_state.model = AutoModelClass.from_pretrained(model_path, **model_args).to('cuda')
     global_state.tokenizer = AutoTokenizer.from_pretrained(tokenizer_path, token=os.environ['hf_token'])
     gc.collect()
     if with_extra_components: