Hugging Face's logo

Spaces:
fruk19
/
nutritionfact_poc
Runtime error

App Files Community
fruk19 commited on
Commit
81657e2
•
1 Parent(s): c21a387

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -10,7 +10,6 @@ PATH_MODEL = os.environ.get('MODEL_REPO_ID')
10
  token = os.environ.get("HF_TOKEN")
11
  task_prompt = os.environ.get("taskpromt")
12
 
13
- # PATH_MODEL = "fruk19/donut_nfact_v4"
14
  processor = DonutProcessor.from_pretrained(PATH_MODEL, token=token)
15
  model = VisionEncoderDecoderModel.from_pretrained(PATH_MODEL, token=token)
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
10
  token = os.environ.get("HF_TOKEN")
11
  task_prompt = os.environ.get("taskpromt")
12
 
 
13
  processor = DonutProcessor.from_pretrained(PATH_MODEL, token=token)
14
  model = VisionEncoderDecoderModel.from_pretrained(PATH_MODEL, token=token)
15
  device = "cuda" if torch.cuda.is_available() else "cpu"