Upload folder using huggingface_hub

app.py

@@ -8,7 +8,12 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-run_llm=run_glm
+if os.getenv("OPENAI_API_KEY",False):
+    print ("using OPENAI_API_KEY")
+    run_llm=run_openai
+else:
+    print ("using glm from huggingface")
+    run_llm=run_glm
 
 PROMPT_STYLE = get_prompt("PROMPT_STYLE")
 PROMPT_TEXT = get_prompt("PROMPT_TEXT")
@@ -19,7 +24,7 @@ SYSPROMPT_TEXT = get_prompt("SYSPROMPT_TEXT")
 def get_song_output(artist, song_text):
     try:
         # Get style music from artist
-        style_glm = run_openai(f"Describe {artist}", sys_prompt=SYSPROMPT_STYLE)
+        style_glm = run_llm(f"Describe {artist}", sys_prompt=SYSPROMPT_STYLE)
         if style_glm:
             style = style_glm
         else:
@@ -29,7 +34,7 @@ def get_song_output(artist, song_text):
         style = ''
 
     try:
-        text_glm = run_openai(f"{PROMPT_TEXT}\n{style}\n#Text song:\n{song_text}\n{PROMPT_TEXT2}", sys_prompt=SYSPROMPT_TEXT)
+        text_glm = run_llm(f"{PROMPT_TEXT}\n{style}\n#Text song:\n{song_text}\n{PROMPT_TEXT2}", sys_prompt=SYSPROMPT_TEXT)
         if text_glm:
             text = text_glm
         else:
@@ -101,7 +106,7 @@ with gr.Blocks() as demo:
         fn=get_song_output,
         inputs=[name_input, text_input],
         outputs=[style_output, song_output]
-    )
+    )   
 
 demo.css = """
 #header {