Upload app.py

app.py

@@ -0,0 +1,112 @@
+#!/usr/bin/env python
+
+from __future__ import annotations
+
+import gradio as gr
+
+# from model import AppModel
+
+MAINTENANCE_NOTICE='Sorry, due to computing resources issues, this space is under maintenance, and will be restored as soon as possible. '
+
+DESCRIPTION = '''# <a href="https://github.com/THUDM/CogVideo">CogVideo</a>
+
+Currently, this Space only supports the first stage of the CogVideo pipeline due to hardware limitations.
+
+The model accepts only Chinese as input.
+By checking the "Translate to Chinese" checkbox, the results of English to Chinese translation with [this Space](https://huggingface.co/spaces/chinhon/translation_eng2ch) will be used as input.
+Since the translation model may mistranslate, you may want to use the translation results from other translation services.
+'''
+NOTES = 'This app is adapted from <a href="https://github.com/hysts/CogVideo_demo">https://github.com/hysts/CogVideo_demo</a>. It would be recommended to use the repo if you want to run the app yourself.'
+FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=THUDM.CogVideo" />'
+
+import json
+import requests
+
+def post(
+        text,
+        translate,
+        seed,
+        only_first_stage,
+        image_prompt
+        ):
+    url = 'https://ccb8is4fqtofrtdsfjebg.ml-platform-cn-beijing.volces.com/devinstance/di-20221130120908-bhpxq/proxy/6201/cogvideo-s1'
+    headers = {
+            "Content-Type": "application/json; charset=UTF-8",
+            "User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.87 Safari/537.36",
+        }
+
+    data = json.dumps({'text': text,
+                    'translate': translate,
+                    'seed': seed,
+                    'only_first_stage': only_first_stage,
+                    'image_prompt': image_prompt
+                    })
+    r = requests.post(url, data, headers=headers)
+
+    translated_text = r.json()['data']['translated_text']
+    result_video = r.json()['data']['result_video']
+    return translated_text, result_video
+
+def main():
+    only_first_stage = True
+    # model = AppModel(only_first_stage)
+
+    with gr.Blocks(css='style.css') as demo:
+        # gr.Markdown(MAINTENANCE_NOTICE)
+        
+        gr.Markdown(DESCRIPTION)
+
+        with gr.Row():
+            with gr.Column():
+                with gr.Group():
+                    text = gr.Textbox(label='Input Text')
+                    translate = gr.Checkbox(label='Translate to Chinese',
+                                            value=False)
+                    seed = gr.Slider(0,
+                                     100000,
+                                     step=1,
+                                     value=1234,
+                                     label='Seed')
+                    only_first_stage = gr.Checkbox(
+                        label='Only First Stage',
+                        value=only_first_stage,
+                        visible=not only_first_stage)
+                    image_prompt = gr.Image(type="filepath",
+                                            label="Image Prompt",
+                                            value=None)
+                    run_button = gr.Button('Run')
+
+            with gr.Column():
+                with gr.Group():
+                    translated_text = gr.Textbox(label='Translated Text')
+                    with gr.Tabs():
+                        with gr.TabItem('Output (Video)'):
+                            result_video = gr.Video(show_label=False)
+
+        # examples = gr.Examples(
+        #     examples=[['骑滑板的皮卡丘', False, 1234, True,None],
+        #               ['a cat playing chess', True, 1253, True,None]],
+        #     fn=model.run_with_translation,
+        #     inputs=[text, translate, seed, only_first_stage,image_prompt],
+        #     outputs=[translated_text, result_video],
+        #     cache_examples=True)
+
+        gr.Markdown(NOTES)
+        gr.Markdown(FOOTER)
+        print(gr.__version__)
+        run_button.click(fn=post,
+                         inputs=[
+                             text,
+                             translate,
+                             seed,
+                             only_first_stage,
+                             image_prompt
+                         ],
+                         outputs=[translated_text, result_video])
+        print(gr.__version__)
+        
+    demo.launch()
+
+
+if __name__ == '__main__':
+    main()