Update due to my update on the model repo

app.py

@@ -79,8 +79,8 @@ def generate_images(G, args):
  
  #----------------------------------------------------------------------------
 
-def inference(seed, truncation_psi):
-    G = stylegan2.models.load(hf_hub_download("hr16/Gwern-TWDNEv3-pytorch_ckpt", "Gs.pth", use_auth_token=os.environ['MODEL_READING_TOKEN']))
+def inference(seed, truncation_psi, model_version):
+    G = stylegan2.models.load(hf_hub_download("hr16/Gwern-TWDNEv3-pytorch_ckpt", f"{model_version}/Gs.pth", use_auth_token=os.environ['MODEL_READING_TOKEN']))
     G.eval()
     return generate_images(
         G, 
@@ -100,8 +100,18 @@ article = ""
 gr.Interface(
     inference, 
     [
-        gr.Number(precision=0, label="Seed (any-bit-size unsigned int)"), 
-        gr.Slider(0, 2, step=0.1, value=0.7, label='Truncation psi (aka creative level, between 0 and 2)')
+        gr.Number(precision=0, label="PCG64 PRNG Seed (any-bit-size unsigned int, note that it may different from the original site)"), 
+        gr.Slider(0, 2, step=0.1, value=0.7, label='Truncation psi (aka creative level, between 0 and 2)'),
+        gr.Radio(
+            ["iteration-24664", "iteration-18528", "iteration-17325"], 
+            value="iteration-24664", 
+            type="value", 
+            label=[
+                "TWDNEv3 iteration 24664 (best and current version on TWDNE)", 
+                "TWDNEv3 iteration 18528 (the most used version on Internet)", 
+                "TWDNEv3 iteration 17325"
+            ]           
+        )
     ], 
     gr.outputs.Image(type="pil"),
     title=title,description=description,article=article,allow_flagging=False,allow_screenshot=False