piyush2102020 commited on
Commit
6247c8e
·
verified ·
1 Parent(s): d042ee1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ from veronica import Veronica # Import Veronica class
22
  # Load tokenizer and model
23
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
24
  model = Veronica(VeronicaConfig())
25
- weights=torch.load("veronica_super_trained.pth",map_location='cpu')
26
  model.load_state_dict(weights)
27
  model.eval()
28
  firebase_key_path = os.getenv("firebase_db_url")
 
22
  # Load tokenizer and model
23
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
24
  model = Veronica(VeronicaConfig())
25
+ weights=torch.load("veronica_model/veronica_super_trained.pth",map_location='cpu')
26
  model.load_state_dict(weights)
27
  model.eval()
28
  firebase_key_path = os.getenv("firebase_db_url")