Update app.py

app.py

@@ -13,6 +13,7 @@ from constants import css, examples, img_height, img_width, num_images_to_gen
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 from huggingface_hub import from_pretrained_keras
+from huggingface_hub import Repository
 
 import requests
 
@@ -155,6 +156,15 @@ def update_compute_options(provider, region):
         value=avalialbe_compute_options[0] if len(avalialbe_compute_options) > 0 else None
     )
 
+def get_sha_of_branch(model_id, branch_name):
+    repo = Repository(
+        local_dir=model_id,
+        clone_from=model_id,
+        repo_type="model",
+        skip_lfs_files=True,
+        revision=branch_name)
+
+    return repo.git_head_hash()
 
 with gr.Blocks() as demo2:
     gr.Markdown(
@@ -170,6 +180,10 @@ with gr.Blocks() as demo2:
     )    
         
     providers = avaliable_providers()
+    head_sha = get_sha_of_branch(
+        "chansung/my-funny-cat",
+        "v1673257770"
+    )
     
     with gr.Row():
         provider_selector = gr.Dropdown(
@@ -188,16 +202,14 @@ with gr.Blocks() as demo2:
         provider_selector.change(update_regions, inputs=provider_selector, outputs=region_selector)
 
     with gr.Row():
-        repository_selector = gr.Dropdown(
-            choices=["my-funny-cat"],
+        repository_selector = gr.Textbox(
             value="my-funny-cat",
             interactive=False,
             label="target repository"
         )
 
-        repository_selector = gr.Dropdown(
-            choices=["v1673257770"],
-            value="v1673257770",
+        repository_selector = gr.Textbox(
+            value=f"v1673257770/{head_sha[:7]}",
             interactive=False,
             label="model version(branch)"
         )