Spaces:
Runtime error
Runtime error
piyush2102020
commited on
Update app.py
Browse files
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")
|