Update app.py

app.py

@@ -12,14 +12,15 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoDistributedModelForCausalLM.from_pretrained(model_name)
 
 # Run the model as if it were on your computer
-def chat(id, npc, text):
+def chat2(id, npc, text):
     prom = ""
     inputs = tokenizer(prom, return_tensors="pt")["input_ids"]
     outputs = model.generate(inputs, max_new_tokens=100)
     print(tokenizer.decode(outputs[0]))
 
     return text
-
+def chat(id, npc, text):
+    return f"{text}에 대한 {npc}의 응답"