nkasmanoff commited on
Commit
7049830
·
1 Parent(s): f8aa4be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import re
2
  import gradio as gr
3
  from PIL import Image
4
  from transformers import AutoProcessor, AutoModelForCausalLM
5
-
6
  device='cpu'
7
 
8
  processor = AutoProcessor.from_pretrained("microsoft/git-base")
 
2
  import gradio as gr
3
  from PIL import Image
4
  from transformers import AutoProcessor, AutoModelForCausalLM
5
+ from knowledge_extraction import get_entities, get_relations
6
  device='cpu'
7
 
8
  processor = AutoProcessor.from_pretrained("microsoft/git-base")