Update app.py

app.py

@@ -6,8 +6,8 @@ import gradio as gr
 import torch
 import torchaudio
 import os
+import requests
 from transformers import pipeline
-from pixabay import Image, Video
 import tempfile
 
 # 기본 설정
@@ -44,10 +44,7 @@ setup_eval_logging()
 
 # 번역기 및 Pixabay API 설정
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
-
-
-
-import requests
+PIXABAY_API_KEY = "33492762-a28a596ec4f286f84cd328b17"
 
 def search_pixabay_videos(query, api_key):
     base_url = "https://pixabay.com/api/videos/"
@@ -63,15 +60,6 @@ def search_pixabay_videos(query, api_key):
         return [video['videos']['large']['url'] for video in data.get('hits', [])]
     return []
 
-# 메인 코드에서 pixabay 관련 부분 수정
-PIXABAY_API_KEY = "33492762-a28a596ec4f286f84cd328b17"
-
-def search_videos(query):
-    query = translate_prompt(query)
-    return search_pixabay_videos(query, PIXABAY_API_KEY)
-
-
-
 # CSS 스타일 정의
 custom_css = """
 .gradio-container {
@@ -150,8 +138,7 @@ def translate_prompt(text):
 
 def search_videos(query):
     query = translate_prompt(query)
-    videos = pixabay_video.search(q=query, per_page=80)
-    return [video.video_large for video in videos['hits']]
+    return search_pixabay_videos(query, PIXABAY_API_KEY)
 
 @spaces.GPU
 @torch.inference_mode()