mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2024-12-21 02:19:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

changes

Browse Source
This commit is contained in:
Abubakar Abid 2020-06-11 14:37:15 -05:00
parent 0528ed67aa
commit 13a82994bc
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gradio/networking.py
View File

@ -140,7 +140,7 @@ def serve_files_in_background(interface, port, directory_to_serve=None):
predictions = [] predictions = []
for predict_fn in interface.predict: for predict_fn in interface.predict:
prediction = predict_fn(*processed_input) prediction = predict_fn(*processed_input)
if len(interface.output_interfaces) == 1: if len(interface.output_interfaces) / len(interface.predict) == 1:
prediction = [prediction] prediction = [prediction]
predictions.extend(prediction) predictions.extend(prediction)
processed_output = [output_interface.postprocess(predictions[i]) for i, output_interface in enumerate(interface.output_interfaces)] processed_output = [output_interface.postprocess(predictions[i]) for i, output_interface in enumerate(interface.output_interfaces)]

1
gradio/outputs.py
View File

@ -126,6 +126,7 @@ class Image(AbstractOutput):
def postprocess(self, prediction): def postprocess(self, prediction):
""" """
""" """
print(prediction.shape)
return preprocessing_utils.encode_array_to_base64(prediction) return preprocessing_utils.encode_array_to_base64(prediction)
def rebuild_flagged(self, dir, msg): def rebuild_flagged(self, dir, msg):