Spaces:

UB2002
/

email_classification

Runtime error

UB2002

final commmits

a08e3d4 over 1 year ago

339 Bytes

import joblib
from typing import Tuple

# Load saved pipeline
try:
    pipeline = joblib.load('model.joblib')
except FileNotFoundError:
    raise RuntimeError("file not found error")

def classify_email(text: str) -> str:
    """
    Given masked email text, returns predicted category label.
    """
    return pipeline.predict([text])[0]

	import joblib
	from typing import Tuple

	# Load saved pipeline
	try:
	pipeline = joblib.load('model.joblib')
	except FileNotFoundError:
	raise RuntimeError("file not found error")

	def classify_email(text: str) -> str:
	"""
	Given masked email text, returns predicted category label.
	"""
	return pipeline.predict([text])[0]