academy04 commited on
Commit
1ae841c
·
1 Parent(s): e69391e

fix length data bug

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. test_script/summarize_text.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ import os
4
  import chromadb
5
  from datasets import load_dataset
6
 
7
- ds = load_dataset("RTVS/SpotifyLyrics001")[:500]
8
  apikey=os.getenv('apikey')
9
  client = OpenAI(
10
  base_url="https://openrouter.ai/api/v1",
 
4
  import chromadb
5
  from datasets import load_dataset
6
 
7
+ ds = load_dataset("RTVS/SpotifyLyrics001")
8
  apikey=os.getenv('apikey')
9
  client = OpenAI(
10
  base_url="https://openrouter.ai/api/v1",
test_script/summarize_text.py CHANGED
@@ -1,7 +1,7 @@
1
  from datasets import load_dataset
2
-
3
- ds = load_dataset("RTVS/SpotifyLyrics001")
4
  from openai import OpenAI
 
 
5
 
6
  client = OpenAI(
7
  base_url="https://openrouter.ai/api/v1",
 
1
  from datasets import load_dataset
 
 
2
  from openai import OpenAI
3
+ ds = load_dataset("RTVS/SpotifyLyrics001")
4
+
5
 
6
  client = OpenAI(
7
  base_url="https://openrouter.ai/api/v1",