updated weights for first rev model

app.py

@@ -42,6 +42,14 @@ gpt_rev_weights_path = huggingface_hub.hf_hub_download(
     "jefsnacker/surname_generator",
     "rev_gpt_weights.pt")
 
+gpt_first_rev_config_path = huggingface_hub.hf_hub_download(
+    "jefsnacker/surname_generator",
+    "first_name_gpt_config.yaml")
+
+gpt_first_rev_weights_path = huggingface_hub.hf_hub_download(
+    "jefsnacker/surname_generator",
+    "first_name_gpt_weights.pt")
+
 with open(mlp_config_path, 'r') as file:
     mlp_config = yaml.safe_load(file)
 
@@ -54,6 +62,9 @@ with open(gpt_micro_config_path, 'r') as file:
 with open(gpt_rev_config_path, 'r') as file:
     gpt_rev_config = yaml.safe_load(file)
 
+with open(gpt_first_rev_config_path, 'r') as file:
+    gpt_first_rev_config = yaml.safe_load(file)
+
 ##################################################################################
 ## MLP
 ##################################################################################
@@ -329,6 +340,10 @@ gpt_rev = GPT(gpt_rev_config)
 gpt_rev.load_state_dict(torch.load(gpt_rev_weights_path))
 gpt_rev.eval()
 
+gpt_first_rev = GPT(gpt_first_rev_config)
+gpt_first_rev.load_state_dict(torch.load(gpt_first_rev_weights_path))
+gpt_first_rev.eval()
+
 ##################################################################################
 ## Gradio App
 ##################################################################################
@@ -351,9 +366,8 @@ def generate_names(name_start, name_end, number_of_names, model):
         config = gpt_rev_config
         sample_fcn = gpt_rev.sample_char
     elif model == "GPT First Rev":
-        # TODO: Change model!
-        config = gpt_rev_config
-        sample_fcn = gpt_rev.sample_char
+        config = gpt_first_rev_config
+        sample_fcn = gpt_first_rev.sample_char
     else:
         return "Error: Model not selected"