Update app.py

app.py

@@ -1,14 +1,17 @@
 import gradio as gr
+from transformers import pipeline
 
-def get_pipe():
-    from transformers import AutoTokenizer, AutoModelForCausalLM
-    model_name = "heegyu/koalpaca-355m"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    tokenizer.truncation_side = "right"
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-    return model, tokenizer
+sentiment = pipeline("sentiment-analysis")
 
-def get_response(tokenizer, model, context):
+def get_sentiment(입력):
+    # from transformers import AutoTokenizer, AutoModelForCausalLM
+    # model_name = "heegyu/koalpaca-355m"
+    # tokenizer = AutoTokenizer.from_pretrained(model_name)
+    # tokenizer.truncation_side = "right"
+    # model = AutoModelForCausalLM.from_pretrained(model_name)
+    return sentiment(입력)
+
+def get_response(output):
     context = f"<usr>{context}\n<sys>"
     inputs = tokenizer(
         context,