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()
#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)