Update

README.md

@@ -4,7 +4,7 @@ emoji: 📈
 colorFrom: green
 colorTo: indigo
 sdk: gradio
-sdk_version: 3.19.1
+sdk_version: 3.36.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -6,23 +6,17 @@ import gradio as gr
 
 from model import Model
 
-DESCRIPTION = '''# MangaLineExtraction_PyTorch
-
-This is an unofficial demo for [https://github.com/ljsabc/MangaLineExtraction_PyTorch](https://github.com/ljsabc/MangaLineExtraction_PyTorch).
-'''
+DESCRIPTION = '# [MangaLineExtraction_PyTorch](https://github.com/ljsabc/MangaLineExtraction_PyTorch)'
 
 model = Model()
 
 with gr.Blocks(css='style.css') as demo:
     gr.Markdown(DESCRIPTION)
-
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(label='Input', type='numpy')
             run_button = gr.Button(value='Run')
         with gr.Column():
             result = gr.Image(label='Result', type='numpy', elem_id='result')
-
     run_button.click(fn=model.predict, inputs=input_image, outputs=result)
-
-demo.queue().launch(show_api=False)
+demo.queue().launch()

model.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import os
 import pathlib
 import sys
 
@@ -16,8 +15,6 @@ sys.path.insert(0, submodule_dir.as_posix())
 
 from model_torch import res_skip
 
-HF_TOKEN = os.getenv('HF_TOKEN')
-
 MAX_SIZE = 1000
 
 
@@ -29,9 +26,7 @@ class Model:
 
     def _load_model(self) -> nn.Module:
         ckpt_path = huggingface_hub.hf_hub_download(
-            'hysts/MangaLineExtraction_PyTorch',
-            'erika.pth',
-            use_auth_token=HF_TOKEN)
+            'public-data/MangaLineExtraction_PyTorch', 'erika.pth')
         state_dict = torch.load(ckpt_path)
         model = res_skip()
         model.load_state_dict(state_dict)