Update README.md

This commit is contained in:
Daya Guo 2023-10-26 18:49:01 +08:00 committed by GitHub
parent 8ba4cdcb63
commit e3c9b42f86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -135,9 +135,9 @@ class IrisClassifier(nn.Module):
return predicted.numpy() return predicted.numpy()
#main.py #main.py
from utils import load_data, evaluate_predictions 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 # Model training and evaluation
""" """
inputs = tokenizer.encode(input_text, return_tensors="pt").to(device) inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)