Update app.py

app.py

@@ -13,13 +13,14 @@ def inference(question: str) -> str:
 
     result = tokenizer.decode(token_ids=outputs[0], skip_special_tokens=True)
 
-    print("question:", question, "answer:", result)
+    # print("question:", question, "answer:", result)
     return result
 
-test_id = 1160
-# 거실조명1 꺼주세요	"Home:거실,NickName:조명1,ActionID:1061,Value:없음"
-print("model result:", inference(dataset_finetuning["train"][test_id]["Q"]))
-print("real result:", dataset_finetuning["train"][test_id]["C"])
+# test_id = 1160
+# # 거실조명1 꺼주세요	"Home:거실,NickName:조명1,ActionID:1061,Value:없음"
+# print("model result:", inference(dataset_finetuning["train"][test_id]["Q"]))
+# print("real result:", dataset_finetuning["train"][test_id]["C"])
+print("거실 조명1 꺼주세요")
 
 gr.load("models/yeye776/t5-OndeviceAI-HomeIoT").launch()
 # iface = gradio.Interface(fn=pipe_home, inputs="text", outputs="text")