Spaces:

janrudolf
/

text-summarizer

Sleeping

janrudolf commited on Mar 4

Commit

f6cbcb4

1 Parent(s): 0c27cd6

Use HF uploaded model

Files changed (1) hide show

api.py CHANGED Viewed

@@ -64,7 +64,7 @@ async def lifespan(app: FastAPI):
         print(f"--- GPU Name: {torch.cuda.get_device_name(0)} ---")
     # LOAD MODEL (local directory; tokenizer from same path for offline/consistency)
-    model_id = "model/bart-finetuned-wobbly-bush-56-h9vtgsw1"
     ml_models["tokenizer"] = AutoTokenizer.from_pretrained(model_id)
     ml_models["model"] = AutoModelForSeq2SeqLM.from_pretrained(model_id).to(device)
     ml_models["device"] = device

         print(f"--- GPU Name: {torch.cuda.get_device_name(0)} ---")
     # LOAD MODEL (local directory; tokenizer from same path for offline/consistency)
+    model_id = "janrudolf/bart-finetuned-wobbly-bush-56-h9vtgsw1"
     ml_models["tokenizer"] = AutoTokenizer.from_pretrained(model_id)
     ml_models["model"] = AutoModelForSeq2SeqLM.from_pretrained(model_id).to(device)
     ml_models["device"] = device