Update app.py

app.py

@@ -10,7 +10,8 @@ from PIL import Image
 @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, re.Pattern: lambda _: None}, allow_output_mutation=True, suppress_st_warning=True)
 def get_model(model_name, model_path='pytorch_model.bin'):
     tokenizer = transformers.GPT2Tokenizer.from_pretrained(model_name)
-    model = transformers.OPTForCausalLM.from_pretrained('big-kek/NeuroSkeptic', device_map='cpu')
+    model = transformers.OPTForCausalLM.from_pretrained(model_name)
+    model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
     
     model.eval()
     return model, tokenizer