Update app.py

app.py

@@ -84,9 +84,15 @@ def inference(prompt):
 
     cos_df["word"].values
 
-    outputs = " ".join(["#" + s for s in cos_df["word"].values])
-
-    outputs
+    # 명사만 추출
+    outputs = []
+    for word in cos_df["word"].values:
+        okt = Okt()
+        tokenized_doc = okt.pos(word)
+        tokenized_nouns = ' '.join([word[0] for word in tokenized_doc if word[1] == 'Noun'])
+        outputs.append("#" + tokenized_nouns)
+
+    outputs = " ".join(outputs)
 
     return outputs