update theme for slider

app.py

@@ -11,41 +11,28 @@ model_glm = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True
 model_glm = model_glm.eval()
 
 # Load pre-trained model and tokenizer for Chinese to English translator
-from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
-model_chtoen = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
-tokenizer_chtoen = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
+#from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
+#model_chtoen = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
+#tokenizer_chtoen = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
 
-def translate_Chinese_English(chinese_text): 
-    # translate Chinese to English
-    tokenizer_chtoen.src_lang = "zh"
-    encoded_zh = tokenizer_chtoen(chinese_text, return_tensors="pt")
-    generated_tokens = model_chtoen.generate(**encoded_zh, forced_bos_token_id=tokenizer_chtoen.get_lang_id("en"))
-    trans_eng_text = tokenizer_chtoen.batch_decode(generated_tokens, skip_special_tokens=True)
-    return trans_eng_text[0]
 
     
 # Define function to generate model predictions and update the history
-def predict_glm_stream(input, top_p, temperature, history=[]): #, top_p, temperature):
-    print(f"1 outside for loop OG history is ^^- {history}")
+def predict_glm_stream(input, top_p, temperature, history=[]): 
     history = list(map(tuple, history))
-    print(f"2 outside for loop OG history is ^^- {history}")
-    for response, updates in model_glm.stream_chat(tokenizer_glm, input, history, top_p=top_p, temperature=temperature):   #history
-        print(f"In for loop resonse is ^^- {response}")
-        print(f"In for loop updates is ^^- {updates}")
-        # translate Chinese to English
-        #history = [(query, translate_Chinese_English(response)) for query, response in history]
-        print(f"*******")
-        yield updates #history+updates
-
+    for response, updates in model_glm.stream_chat(tokenizer_glm, input, history, top_p=top_p, temperature=temperature):   
+        yield updates 
     
 def reset_textbox():
     return gr.update(value="")
 
-def reset_chat(chatbot, state):
-    # debug
-    #print(f"^^chatbot value is - {chatbot}")
-    #print(f"^^state value is - {state}")
-    return None, []
+def translate_Chinese_English(chinese_text): 
+    # translate Chinese to English
+    tokenizer_chtoen.src_lang = "zh"
+    encoded_zh = tokenizer_chtoen(chinese_text, return_tensors="pt")
+    generated_tokens = model_chtoen.generate(**encoded_zh, forced_bos_token_id=tokenizer_chtoen.get_lang_id("en"))
+    trans_eng_text = tokenizer_chtoen.batch_decode(generated_tokens, skip_special_tokens=True)
+    return trans_eng_text[0]
 
 
 title = """<h1 align="center"> 🚀CHatGLM-6B - A Streaming Chatbot with Gradio</h1>
@@ -60,7 +47,7 @@ However, due to the small size of ChatGLM-6B, it is currently known to have cons
 theme = gr.themes.Default(#color contructors
                           primary_hue="violet", 
                           secondary_hue="indigo",
-                          neutral_hue="purple") 
+                          neutral_hue="purple").set(slider_color="#800080")
                            
 with gr.Blocks(css="""#col_container {margin-left: auto; margin-right: auto;}
                 #chatglm {height: 520px; overflow: auto;} """, theme=theme ) as demo:
@@ -80,7 +67,7 @@ with gr.Blocks(css="""#col_container {margin-left: auto; margin-right: auto;}
         with gr.Box():
             chatbot_glm = gr.Chatbot(elem_id="chatglm", label='THUDM-ChatGLM6B')
           
-        with gr.Box():
+        with gr.Accordion(label="Parameters for ChatGLM-6B", open=False):
             gr.HTML("Parameters for ChatGLM-6B", visible=True)
             top_p = gr.Slider(minimum=-0, maximum=1.0,value=1, step=0.05,interactive=True, label="Top-p", visible=True)
             temperature = gr.Slider(minimum=-0, maximum=5.0, value=1, step=0.1, interactive=True, label="Temperature", visible=True)
@@ -95,7 +82,6 @@ with gr.Blocks(css="""#col_container {margin-left: auto; margin-right: auto;}
                 [chatbot_glm],)
     b1.click(reset_textbox, [], [inputs])
 
-    #b2.click(reset_chat, [chatbot_glm, state_glm], [chatbot_glm, state_glm])
     b2.click(lambda: None, None, chatbot_glm, queue=False)
     
     gr.HTML('''<center><a href="https://huggingface.co/spaces/ysharma/ChatGLM-6b_Gradio_Streaming?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>Duplicate the Space and run securely with your OpenAI API Key</center>''')