Update app.py

app.py

@@ -1,19 +1,16 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
-# 모델 로드
 model_name = "Yuchan5386/NaturaAI-1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# 텍스트 생성 함수
 def generate_text(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(**inputs)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-# Gradio 인터페이스 생성, api=True로 API 엔드포인트 활성화
-iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", api=True)
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 
-# Gradio 인터페이스 실행
-iface.launch(share=True)
+# API로 호출할 수 있도록 설정
+iface.launch(share=True, api=True)