From e3c9b42f867db5483fe69430030dfa9dc28f8b72 Mon Sep 17 00:00:00 2001
From: Daya Guo <40300434+guoday@users.noreply.github.com>
Date: Thu, 26 Oct 2023 18:49:01 +0800
Subject: [PATCH] Update README.md

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 4167a3e..8979d31 100644
--- a/README.md
+++ b/README.md
@@ -135,9 +135,9 @@ class IrisClassifier(nn.Module):
         return predicted.numpy()
 #main.py
 from utils import load_data, evaluate_predictions
-from model import IrisClassifier
+from model import IrisClassifier as Classifier
 
-def pipeline():
+def main():
     # Model training and evaluation
 """
 inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)