Update app.py

app.py

@@ -13,6 +13,7 @@ from veronica import Veronica, VeronicaConfig
 import firebase_admin
 import json
 from firebase_admin import credentials, db
+import torch
 import time
 from veronica import Veronica  # Import Veronica class
 
@@ -21,6 +22,8 @@ from veronica import Veronica  # Import Veronica class
 # Load tokenizer and model
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 model = Veronica(VeronicaConfig())
+weights=torch.load("veronica_super_trained.pth",map_location='cpu')
+model.load_state_dict(weights)
 model.eval()
 firebase_key_path = os.getenv("firebase_db_url")
 cred = json.load(os.getenv("firebase_credentials"))