sayyedAhmed
/

Crisis_Severity_Predictor_LSTM

Time Series Forecasting

Inference Endpoints

Model card Files Files and versions Community

sayyedAhmed commited on Nov 23, 2024

Commit

65a224a

•

1 Parent(s): 744b965

saved changed

Files changed (1) hide show

inference.py +1 -1

inference.py CHANGED Viewed

@@ -29,7 +29,7 @@ from transformers import pipeline
 model_name = "sayyedAhmed/Crisis_Severity_Predictor_LSTM"
 # Create the pipeline with manual framework specification (using 'tf' for TensorFlow)
-classifier = pipeline("text-classification", model=model_name, framework="tf")
 # Use the pipeline to run inference
 result = classifier("Example text for classification.")

 model_name = "sayyedAhmed/Crisis_Severity_Predictor_LSTM"
 # Create the pipeline with manual framework specification (using 'tf' for TensorFlow)
+classifier = pipeline("text-classification", model=model_name, framework="pt")
 # Use the pipeline to run inference
 result = classifier("Example text for classification.")