.

README.md

@@ -4,7 +4,7 @@ emoji: 🐢
 colorFrom: purple
 colorTo: gray
 sdk: gradio
-sdk_version: 4.42.0
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -4,14 +4,13 @@ import argparse
 import gradio as gr
 import spaces
 
-# from utils import Image2Text
+from utils import Image2Text
 
 @spaces.GPU(duration=60)
 def predict(input_img):
-    return {'0': 'hellow', '1': 'world'}
-    # global image_to_text
-    # contents = image_to_text.get_text(input_img, num_beams=4)
-    # return contents
+    global image_to_text
+    contents = image_to_text.get_text(input_img, num_beams=4)
+    return contents
 
 def get_image_path_list(folder_name):
     image_basename_list = os.listdir(folder_name)
@@ -90,11 +89,11 @@ css = """
 """
 
 if __name__ == "__main__":
-    # repo_id = os.getenv('MODEL_REPO_ID')
-    # hf_token = os.environ.get("HF_TOKEN")
+    repo_id = os.getenv('MODEL_REPO_ID')
+    hf_token = os.environ.get("HF_TOKEN")
 
-    # device = "cuda"
-    # image_to_text = Image2Text(repo_id, hf_token=hf_token, device=device)
+    device = "cuda"
+    image_to_text = Image2Text(repo_id, hf_token=hf_token, device=device)
 
     with gr.Blocks(css=css) as demo:
         gr.Markdown(title)