toilaluan commited on
Commit
3f07cae
·
1 Parent(s): 62786d6
Files changed (2) hide show
  1. app.py +1 -0
  2. scorer.py +0 -1
app.py CHANGED
@@ -44,6 +44,7 @@ def draw_colored_graph(dependencies, questions, answers):
44
 
45
  processor = DSGPromptProcessor("mistralai/Mixtral-8x7B-Instruct-v0.1")
46
 
 
47
  def process_image(image, prompt):
48
  tuples, _ = processor.generate_tuples(prompt)
49
  dependencies, _ = processor.generate_dependencies(tuples)
 
44
 
45
  processor = DSGPromptProcessor("mistralai/Mixtral-8x7B-Instruct-v0.1")
46
 
47
+ @spaces.GPU()
48
  def process_image(image, prompt):
49
  tuples, _ = processor.generate_tuples(prompt)
50
  dependencies, _ = processor.generate_dependencies(tuples)
scorer.py CHANGED
@@ -319,7 +319,6 @@ Each tuple contains the following information:
319
 
320
  return sorted_questions
321
 
322
- @spaces.GPU()
323
  def get_reward(
324
  self,
325
  questions: list[str],
 
319
 
320
  return sorted_questions
321
 
 
322
  def get_reward(
323
  self,
324
  questions: list[str],