Scharbhen commited on
Commit
3f1d188
1 Parent(s): 2f21f91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ login(os.getenv('hf_token'))
12
  def infer_ocrvqa(image, question):
13
  model = PaliGemmaForConditionalGeneration.from_pretrained("google/paligemma-3b-ft-ocrvqa-896").to("cuda")
14
  processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-ft-ocrvqa-896")
15
- inputs = processor(images=image,text=question, return_tensors="pt").to("cuda")
 
16
  predictions = model.generate(**inputs, max_new_tokens=100)
17
  return processor.decode(predictions[0], skip_special_tokens=True)[len(question):].lstrip("\n")
18
 
 
12
  def infer_ocrvqa(image, question):
13
  model = PaliGemmaForConditionalGeneration.from_pretrained("google/paligemma-3b-ft-ocrvqa-896").to("cuda")
14
  processor = PaliGemmaProcessor.from_pretrained("google/paligemma-3b-ft-ocrvqa-896")
15
+ systemprompt = "Ты ассистент по анализу финансовых отчетов. Ниже приведены вопросы по данным на изображении. Необходимо отвечать на вопросы по суммам в таблицах максимально точно и обращать внимание на названия колонок таблиц. Вопросы: "
16
+ inputs = processor(images=image,text=systemprompt+question, return_tensors="pt").to("cuda")
17
  predictions = model.generate(**inputs, max_new_tokens=100)
18
  return processor.decode(predictions[0], skip_special_tokens=True)[len(question):].lstrip("\n")
19