m

app.py

@@ -1,7 +1,16 @@
 from fastapi import FastAPI
+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!"]
+
+tokens = tokenizer(sequences, padding=True, truncation=True, return_tensors="pt")
+output = model(**tokens)
 app = FastAPI()
 
 @app.get("/")
 def greet_json():
-    return {"Hello": "World!"}
+    return {"Hello": output}

requirements.txt

@@ -1,2 +1,4 @@
 fastapi
 uvicorn[standard]
+torch
+transformers