morethanair commited on
Commit
13f20c2
·
1 Parent(s): 48c0344

로컬 파일로 app.py 강제 overwrite (MCP)

Browse files
Files changed (1) hide show
  1. app.py +5 -14
app.py CHANGED
@@ -1,21 +1,12 @@
1
  import streamlit as st
2
  import os
3
  from pinecone import Pinecone
4
- from typing import List, Dict
5
- import logging
6
-
7
- # Compatibility shim for older huggingface_hub versions
8
- try:
9
- from huggingface_hub import cached_download
10
- except Exception:
11
- from huggingface_hub import hf_hub_download as cached_download
12
- import huggingface_hub
13
- huggingface_hub.cached_download = hf_hub_download
14
-
15
  from sentence_transformers import SentenceTransformer
 
16
  import re # For parsing timestamp and extracting video ID
17
  import streamlit.components.v1 as components # For embedding HTML
18
  from openai import OpenAI # Import OpenAI library
 
19
 
20
  # Setup logging
21
  logging.basicConfig(level=logging.INFO)
@@ -93,7 +84,7 @@ def generate_youtube_embed_html(youtube_url: str, timestamp: str) -> str | None:
93
  PINECONE_API_KEY = os.getenv("PINECONE_API_KEY","pcsk_PZHLK_TRAvMCyNmJM4FKGCX7rbbY22a58fhnWYasx1mf3WL6sRasoASZXfsbnJYvCQ13w") # Load from environment variable
94
  PINECONE_ENV = os.getenv("PINECONE_ENV", "us-east-1")
95
  INDEX_NAME = "video-embeddings"
96
- EMBEDDING_MODEL_ID = 'jhgan/ko-sroberta-multitask' # 모델 ID로 변경
97
 
98
  # OpenAI 설정
99
  OPENAI_API_KEY = "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"
@@ -118,8 +109,8 @@ def init_pinecone():
118
  def load_embedding_model():
119
  """Sentence Transformer 모델을 로드합니다."""
120
  try:
121
- model = SentenceTransformer(EMBEDDING_MODEL_ID) # 수정된 모델 ID 사용
122
- logger.info(f"Successfully loaded embedding model: {EMBEDDING_MODEL_ID}") # 수정된 모델 ID 사용
123
  return model
124
  except Exception as e:
125
  st.error(f"임베딩 모델 로딩 중 오류 발생: {e}")
 
1
  import streamlit as st
2
  import os
3
  from pinecone import Pinecone
 
 
 
 
 
 
 
 
 
 
 
4
  from sentence_transformers import SentenceTransformer
5
+ from typing import List, Dict
6
  import re # For parsing timestamp and extracting video ID
7
  import streamlit.components.v1 as components # For embedding HTML
8
  from openai import OpenAI # Import OpenAI library
9
+ import logging
10
 
11
  # Setup logging
12
  logging.basicConfig(level=logging.INFO)
 
84
  PINECONE_API_KEY = os.getenv("PINECONE_API_KEY","pcsk_PZHLK_TRAvMCyNmJM4FKGCX7rbbY22a58fhnWYasx1mf3WL6sRasoASZXfsbnJYvCQ13w") # Load from environment variable
85
  PINECONE_ENV = os.getenv("PINECONE_ENV", "us-east-1")
86
  INDEX_NAME = "video-embeddings"
87
+ EMBEDDING_MODEL = "jhgan/ko-sroberta-multitask"
88
 
89
  # OpenAI 설정
90
  OPENAI_API_KEY = "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"
 
109
  def load_embedding_model():
110
  """Sentence Transformer 모델을 로드합니다."""
111
  try:
112
+ model = SentenceTransformer(EMBEDDING_MODEL)
113
+ logger.info(f"Successfully loaded embedding model: {EMBEDDING_MODEL}")
114
  return model
115
  except Exception as e:
116
  st.error(f"임베딩 모델 로딩 중 오류 발생: {e}")