llama3

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+.env

app.py

@@ -2,9 +2,12 @@ from fastapi import FastAPI
 from fastapi.responses import RedirectResponse
 from transformers import pipeline
 import torch
-from huggingface_hub import hf_hub_download
+from huggingface_hub import login
+import os
 
-hf_hub_download(repo_id="google/pegasus-xsum", filename="config.json")
+access_token = os.getenv("HF_ACCESS_TOKEN", "")
+
+login(token=access_token)
 
 # Create a new FastAPI app instance
 app = FastAPI()