Update app.py

app.py

@@ -29,6 +29,8 @@ import requests
 # # Warm-up the model.
 # _ = model.text_to_image("Teddy bear", batch_size=num_images_to_gen)
 
+head_sha = "398e79c789669981a2ab1da1fbdafc3998c7b08a"
+
 def generate_image_fn(prompt: str, unconditional_guidance_scale: int) -> list:
     start_time = time.time()
     # `images is an `np.ndarray`. So we convert it to a list of ndarrays.
@@ -163,7 +165,6 @@ def submit(
     provider_selector,
     region_selector,
     repository_selector,
-    revision_selector,
     task_selector,
     framework_selector,
     compute_selector,
@@ -176,7 +177,7 @@ def submit(
 
     size_l_index = compute_resources[0].index("[")
     size_r_index = compute_resources[0].index("]")
-    size = compute_resources[0][size_l_index : size_r_index+1].strip()
+    size = compute_resources[0][size_l_index-1 : size_r_index].strip()
 
     type = compute_resources[-1].strip()
     
@@ -197,7 +198,7 @@ def submit(
           "huggingface": {}
         },
         "repository": repository_selector.lower(),
-        "revision": revision_selector.split("/")[1].lower(),
+        "revision": head_sha,
         "task": task_selector.lower()
       },
       "name": endpoint_name_input,
@@ -233,7 +234,6 @@ with gr.Blocks() as demo2:
     )    
         
     providers = avaliable_providers()
-    head_sha = "398e79c789669981a2ab1da1fbdafc3998c7b08a"
 
     with gr.Row():
         gr.Markdown("""    
@@ -360,7 +360,6 @@ with gr.Blocks() as demo2:
             provider_selector,
             region_selector,
             repository_selector,
-            revision_selector,
             task_selector,
             framework_selector,
             compute_selector,