Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +4 -3
src/streamlit_app.py
CHANGED
@@ -15,10 +15,11 @@ from langchain_community.tools.tavily_search import TavilySearchResults
|
|
15 |
|
16 |
# Set API Keys
|
17 |
os.environ["TOGETHER_API_KEY"] = "5c22e5f0d9af71d1cd7dfac4284fcde8260ca7db9c81a678387c74d0679da268"
|
18 |
-
|
|
|
19 |
|
20 |
# LLM and Embeddings
|
21 |
-
llm = ChatTogether(model="
|
22 |
embeddings = TogetherEmbeddings(model="togethercomputer/m2-bert-80M-32k-retrieval")
|
23 |
|
24 |
# Prompts
|
@@ -80,7 +81,7 @@ def extract_video_id(url):
|
|
80 |
def get_transcript(video_id):
|
81 |
try:
|
82 |
transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=["en"])
|
83 |
-
return " ".join(chunk[
|
84 |
except TranscriptsDisabled:
|
85 |
return None
|
86 |
|
|
|
15 |
|
16 |
# Set API Keys
|
17 |
os.environ["TOGETHER_API_KEY"] = "5c22e5f0d9af71d1cd7dfac4284fcde8260ca7db9c81a678387c74d0679da268"
|
18 |
+
|
19 |
+
os.environ["TAVILY_API_KEY"] = "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"
|
20 |
|
21 |
# LLM and Embeddings
|
22 |
+
llm = ChatTogether(model="deepseek-ai/DeepSeek-R1-Distill-Llama-70B-free", temperature=0.2)
|
23 |
embeddings = TogetherEmbeddings(model="togethercomputer/m2-bert-80M-32k-retrieval")
|
24 |
|
25 |
# Prompts
|
|
|
81 |
def get_transcript(video_id):
|
82 |
try:
|
83 |
transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=["en"])
|
84 |
+
return " ".join([chunk['text'] for chunk in transcript_list])
|
85 |
except TranscriptsDisabled:
|
86 |
return None
|
87 |
|