toilaluan commited on
Commit
5f84a74
1 Parent(s): 25bd6f8
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -0
app.py CHANGED
@@ -42,7 +42,7 @@ def draw_colored_graph(dependencies, questions, answers):
42
  img = Image.open(buf)
43
  return img
44
 
45
- dsg_scorer = DSGPromptProcessor("mistralai/Mixtral-8x7B-Instruct-v0.1")
46
 
47
  def process_image(image, prompt):
48
  tuples, _ = processor.generate_tuples(prompt)
 
42
  img = Image.open(buf)
43
  return img
44
 
45
+ processor = DSGPromptProcessor("mistralai/Mixtral-8x7B-Instruct-v0.1")
46
 
47
  def process_image(image, prompt):
48
  tuples, _ = processor.generate_tuples(prompt)
requirements.txt CHANGED
@@ -5,3 +5,5 @@ torch
5
  pillow
6
  timm
7
  einops
 
 
 
5
  pillow
6
  timm
7
  einops
8
+ networkx
9
+ matplotlib