Update app.py

app.py

@@ -8,7 +8,7 @@ import asyncio
 import random
 import edge_tts
 import tempfile
-from gradio_client import Client
+import cohere
 import gradio as gr
 from pydub import AudioSegment
 from moviepy.editor import AudioFileClip, concatenate_audioclips
@@ -70,9 +70,7 @@ rss_feed = 'https://www.scmp.com/rss/4/feed'
 
 
 apikey = os.environ.get("API_KEY")
-MODEL = os.environ.get("MODEL")
-
-client = Client(MODEL)
+co = cohere.Client(api_key=apikey)
 
 
 # RSS feeds
@@ -182,15 +180,15 @@ async def main(link, peoples="双人"):
     messages = system_prompt + "\n\n\n" + prompt
 
     
-    completion = client.predict(
-        user_message=messages,
-        api_name="/generate_response"
+    completion = co.chat(
+        model="command-r",
+        message=messages
     )
 
 
     print(completion)
     
-    generated_script = extract_content(completion)
+    generated_script = extract_content(completion.text)
 
     #print("Generated Script:"+generated_script)