Update app.py

app.py

@@ -15,7 +15,7 @@ YOUTUBE_API_KEY = "YOUR_YOUTUBE_API_KEY"
 
 def create_client(model_name):
     token = os.getenv("HF_TOKEN")
-    return InferenceClient(model_name=model_name, token=token)
+    return InferenceClient(model=model_name, token=token)
 
 client = create_client("CohereForAI/c4ai-command-r-plus")
 
@@ -135,7 +135,7 @@ def analyze_titles(video_stats_df, n_clusters=5):
 
 def summarize_cluster(cluster_text, cluster_num):
     response = client(inputs=cluster_text)
-    summary = response['generated_text'].strip()
+    summary = response[0]['generated_text'].strip()
     return summary
 
 def main(query, max_results, period, page, n_clusters=5):
@@ -166,10 +166,10 @@ iface = gr.Interface(
     fn=main,
     inputs=[
         gr.components.Textbox(label="검색 쿼리"),
-        gr.components.Slider(minimum=1, maximum=1000, value=5, label="최대 결과 수"),
+        gr.components.Textbox(label="최대 결과 수", type="number"),
         gr.components.Dropdown(["1주일", "1개월", "3개월"], label="기간"),
         gr.components.Dropdown(["Video Ranking", "Title Analysis"], label="페이지"),
-        gr.components.Slider(minimum=2, maximum=10, value=5, label="클러스터 수")
+        gr.components.Textbox(label="클러스터 수", type="number")
     ],
     outputs=[
         gr.components.Dataframe(label="결과"),