Spaces:
Runtime error
Runtime error
Commit
·
13f20c2
1
Parent(s):
48c0344
로컬 파일로 app.py 강제 overwrite (MCP)
Browse files
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 |
-
|
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(
|
122 |
-
logger.info(f"Successfully loaded embedding model: {
|
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}")
|