app update

app.py

@@ -73,7 +73,7 @@ class Item(BaseModel):
 @app.post("/predict/", response_model=dict)
 async def predict(item: Item):
     model = TransformerEncoder()
-    model, start_epoch = load_model_to_cpu(model, "model.pth")
+    model = load_model_to_cpu(model, "model.pth")
     tokenizer = Tokenizer.from_file("tokenizer.json")
 
     predict_list = predict_fonk(model=model, device=device, example=item.text, tokenizer=tokenizer)
@@ -81,3 +81,7 @@ async def predict(item: Item):
     #Buraya model'in çıktısı gelecek
     #Çıktı formatı aşağıdaki örnek gibi olacak
     return predict_list
+
+
+if __name__=="__main__":
+    uvicorn.run(app,host="0.0.0.0",port=8000)

model.py

@@ -145,6 +145,5 @@ class TransformerEncoder(nn.Module):
 
 def load_model_to_cpu(model, path="model.pth"):
     checkpoint = torch.load(path, map_location=torch.device('cpu'))
-    model.load_state_dict(checkpoint['model_state_dict'])
-    epoch = checkpoint['epoch']
-    return model, epoch
+    model.load_state_dict(checkpoint)
+    return model