Update for Hugging Face Space deployment

app.py

@@ -1,7 +1,38 @@
 import gradio as gr
-from backend.app import create_interface
+from huggingface_hub import InferenceClient
+import os
 
-demo = create_interface()
+def generate_diary(keywords):
+    client = InferenceClient(
+        model="nlpai-lab/kullm-polyglot-5.8b-v2",
+        token=os.environ.get("HUGGINGFACE_API_KEY")
+    )
+    
+    prompt = f"""다음은 오늘 있었던 일의 요약입니다. 이것을 바탕으로 생생하고 감동적인 일기를 작성해주세요.
+    // ... 기존 프롬프트 내용 ...
+    """
+    
+    parameters = {
+        "max_new_tokens": 768,
+        "temperature": 0.88,
+        "top_p": 0.95,
+        "repetition_penalty": 1.35,
+        "top_k": 50,
+        "do_sample": True,
+        "num_return_sequences": 1
+    }
+    
+    response = client.text_generation(prompt, **parameters)
+    return response
+
+# Gradio 인터페이스 생성
+demo = gr.Interface(
+    fn=generate_diary,
+    inputs=gr.Textbox(label="오늘의 키워드를 입력하세요"),
+    outputs=gr.Textbox(label="생성된 일기"),
+    title="AI 일기 도우미",
+    description="키워드를 입력하면 AI가 일기를 작성해줍니다."
+)
 
 if __name__ == "__main__":
     demo.launch() 

requirements.txt

@@ -0,0 +1,10 @@
+flask==2.3.3
+flask-cors==4.0.0
+requests==2.31.0
+python-dotenv==1.0.0
+werkzeug==2.3.7
+huggingface-hub>=0.17.0
+gradio==4.8.0
+torch
+transformers
+accelerate