Canyu commited on
Commit
9381693
·
1 Parent(s): 93ddd94
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -66,20 +66,14 @@ def process_image_4(image_path, prompt):
66
  inputs = []
67
  for p in prompt:
68
  cur_p = map_prompt[p]
69
- image = Image.open(image_path)
70
-
71
- w, h = image.size
72
 
73
  coor_point = []
74
  point_labels = []
75
-
76
- image = image.resize((768, 768), Image.LANCZOS).convert('RGB')
77
 
78
 
79
  cur_input = {
80
- 'input_images': image,
81
- 'original_size': [[w,h]],
82
- 'target_size': [[768, 768]],
83
  'prompt': [cur_p],
84
  'coor_point': coor_point,
85
  'point_labels': point_labels,
@@ -95,6 +89,7 @@ def inf(image_path, prompt):
95
  inputs = process_image_4(image_path, prompt)
96
  # return None
97
  return client.predict(
 
98
  data=inputs,
99
  api_name="/inf"
100
  )
 
66
  inputs = []
67
  for p in prompt:
68
  cur_p = map_prompt[p]
 
 
 
69
 
70
  coor_point = []
71
  point_labels = []
 
 
72
 
73
 
74
  cur_input = {
75
+ # 'original_size': [[w,h]],
76
+ # 'target_size': [[768, 768]],
 
77
  'prompt': [cur_p],
78
  'coor_point': coor_point,
79
  'point_labels': point_labels,
 
89
  inputs = process_image_4(image_path, prompt)
90
  # return None
91
  return client.predict(
92
+ image=handle_file(image_path),
93
  data=inputs,
94
  api_name="/inf"
95
  )