update app.py

app.py

@@ -1,17 +1,22 @@
-from transformers import pipeline
+import subprocess, os
+
+result = subprocess.run(['pip', 'list'], check=True)
+print(f'pip list_1_ = {result}')
+
+os.system("pip install httpx==0.24.1")
+
+result = subprocess.run(['pip', 'list'], check=True)
+print(f'pip list_2_ = {result}')
+
 import gradio as gr
 import random
 import string
 # import paddlehub as hub 
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 from loguru import logger
 
 from utils import get_tmt_client, getTextTrans_tmt
-import subprocess
-
-result = subprocess.run(['pip', 'list'], check=True)
-print(f'pip list = {result}')
 
 tmt_client = get_tmt_client()
 
@@ -36,9 +41,9 @@ def getTextTrans(text, source='zh', target='en'):
 space_ids = {
             "spaces/stabilityai/stable-diffusion": "SD 2.1",
             "spaces/runwayml/stable-diffusion-v1-5": "SD 1.5",
-            "spaces/stabilityai/stable-diffusion-1": "SD 1.0",
-            "dalle_mini_tab": "Dalle mini",
-            "spaces/IDEA-CCNL/Taiyi-Stable-Diffusion-Chinese": "Taiyi(太乙)",
+            # "spaces/stabilityai/stable-diffusion-1": "SD 1.0",
+            # "dalle_mini_tab": "Dalle mini",
+            # "spaces/IDEA-CCNL/Taiyi-Stable-Diffusion-Chinese": "Taiyi(太乙)",
             }
 
 tab_actions = []
@@ -95,20 +100,25 @@ if do_dreamlike_photoreal:
     tab_titles.append("Dreamlike_2.0")
     thanks_info += "[<a style='display:inline-block' href='https://huggingface.co/dreamlike-art/dreamlike-photoreal-2.0' _blank><font style='color:blue;weight:bold;'>dreamlike-photoreal-2.0</font></a>]"
 
-for space_id in space_ids.keys():
-    print(space_id, space_ids[space_id])
+space_ids_keys = space_ids.keys()
+logger.info(f"load_space__000__{space_ids_keys}")
+for space_id in space_ids_keys:
+    logger.info(f"load_space__{space_id}__{space_ids[space_id]}")
+    tab_title = ''
     try:
-        tab_title = space_ids[space_id]
-        tab_titles.append(tab_title)
+        tab_title = space_ids[space_id]        
         if (tab_title == 'Dalle mini'):
             tab_content = gr.Blocks(elem_id='dalle_mini')
+            tab_titles.append(tab_title)
             tab_actions.append(tab_content)            
         else:
             tab_content = gr.Interface.load(space_id)
-            tab_actions.append(tab_content)
+            tab_titles.append(tab_title)
+            tab_actions.append(tab_content)            
             thanks_info += f"[<a style='display:inline-block' href='https://huggingface.co/{space_id}' _blank><font style='color:blue;weight:bold;'>{tab_title}</font></a>]"
+        logger.info(f"load_sucessful__{space_id}__{tab_title}")
     except Exception as e:
-        logger.info(f"load_fail__{space_id}_{e}")
+        logger.info(f"load_fail__{space_id}__{tab_title}____{e}")
 
 start_work = """async() => {
     function isMobile() {
@@ -174,7 +184,7 @@ start_work = """async() => {
             } else if (tabitems_title[i].innerText.indexOf('Dreamlike') >= 0 && tabitems[i].childNodes[0].children.length > 0) {
                 tabitems[i].childNodes[0].children[0].children[1].style.display='none';
             } else if (tabitems_title[i].innerText.indexOf('Dalle mini') >= 0) {
-                window['dalle_mini_block']=  tabitems[i];
+                window['dalle_mini_block'] = tabitems[i];
             }
         }  
         
@@ -228,10 +238,10 @@ start_work = """async() => {
                                     continue;
                                 }
                                 inputText = null;
-                                if (tabitems_title[i].innerText.indexOf('SD') >= 0) {
+                                if (tabitems_title[i].innerText.indexOf('SD') >= 0 && tabitems[i].childNodes[0].children.length > 0) {
                                     text_value = window['gradioEl'].querySelectorAll('#prompt_work')[0].querySelectorAll('textarea')[0].value;
                                     inputText = tabitems[i].children[0].children[1].children[0].querySelectorAll('.gr-text-input')[0];
-                                } else if (tabitems_title[i].innerText.indexOf('Taiyi') >= 0) {
+                                } else if (tabitems_title[i].innerText.indexOf('Taiyi') >= 0 && tabitems[i].childNodes[0].children.length > 0) {
                                     text_value = window['gradioEl'].querySelectorAll('#prompt_work_zh')[0].querySelectorAll('textarea')[0].value;
                                     inputText = tabitems[i].children[0].children[0].children[1].querySelectorAll('.gr-text-input')[0];
                                 }