override gradio version

app.py

@@ -1,3 +1,9 @@
+
+import base64, os
+# add a command for installing flash-attn
+os.system('pip install flash-attn --no-build-isolation')
+os.system("pip install gradio==4.44.1")
+
 from typing import Optional
 import spaces
 import gradio as gr
@@ -7,7 +13,6 @@ from PIL import Image
 import io
 import re
 
-import base64, os
 from util.utils import check_ocr_box, get_yolo_model, get_caption_model_processor, get_som_labeled_img
 from util.som import MarkHelper, plot_boxes_with_marks, plot_circles_with_marks
 from util.process_utils import pred_2_point, extract_bbox, extract_mark_id
@@ -20,8 +25,7 @@ import torch
 from transformers import AutoModelForCausalLM
 from transformers import AutoProcessor 
 
-# add a command for installing flash-attn
-os.system('pip install flash-attn --no-build-isolation')
+
 
 # Define repository and local directory
 repo_id = "microsoft/OmniParser-v2.0"  # HF repo