handsomeFace786 commited on
Commit
5e04c5e
·
verified ·
1 Parent(s): 871f760

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
- os.environ["STREAMLIT_CONFIG_DIR"] = os.path.join(os.path.dirname(__file__), ".streamlit")
 
19
 
20
  # LLM and Embeddings
21
- llm = ChatTogether(model="meta-llama/Llama-3.3-70B-Instruct-Turbo-Free", temperature=0.2)
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["text"] for chunk in transcript_list)
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