Update app.py

app.py

@@ -7,6 +7,7 @@ import re
 import time
 import random
 import os
+from huggingface_hub import InferenceClient
 
 def setup_session():
     try:
@@ -96,6 +97,18 @@ def clean_text(text):
     text = re.sub(r'\s+', ' ', text).strip()
     return text
 
+def create_client(model_name):
+    return InferenceClient(model_name, token=os.getenv("HF_TOKEN"))
+
+client = create_client("OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5")
+
+def call_api(content, system_message, max_tokens, temperature, top_p):
+    messages = [{"role": "system", "content": system_message}, {"role": "user", "content": content}]
+    random_seed = random.randint(0, 1000000)
+    response = client.chat_completion(messages=messages, max_tokens=max_tokens, temperature=temperature, top_p=top_p, seed=random_seed)
+    modified_text = response.choices[0].message.content
+    return modified_text
+
 def analyze_info(category, topic, references1, references2, references3):
     return f"선택한 카테고리: {category}\n블로그 주제: {topic}\n참고 글1: {references1}\n참고 글2: {references2}\n참고 글3: {references3}"