Update app.py

app.py

@@ -17,11 +17,13 @@
 # iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 # iface.launch()
 
-from transformers import AutoTokenizer
+import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 checkpoint = "distilbert-base-uncased-finetuned-sst-2-english"
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
+sequences = ["I've been waiting for a HuggingFace course my whole life.", "So have I!"]
 
-sequence = "I've been waiting for a HuggingFace course my whole life."
-
-model_inputs = tokenizer(sequence)
+tokens = tokenizer(sequences, padding=True, truncation=True, return_tensors="pt")
+output = model(**tokens)