Spaces:
Running
Running
commit
Browse files
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 |
-
'
|
81 |
-
'
|
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 |
)
|