Fix gradio tool demos #31230
Fix gradio tool demos #31230
Conversation
aymeric-roucher
force-pushed
the
fix-gradio-tool-demos
branch
from
0492b4e
to
24a3889
Compare
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
aymeric-roucher
force-pushed
the
fix-gradio-tool-demos
branch
2 times, most recently
from
c2ecd5b
to
663071c
Compare
| @@ -88,7 +88,8 @@ class AgentImage(AgentType, ImageType): | |||
| """ | |||
|
|
|||
| def __init__(self, value): | |||
| super().__init__(value) | |||
| AgentType.__init__(self, value) | |||
| ImageType.__init__(self) | |||
There was a problem hiding this comment.
I found that without this, the ImageType attributes would not be inherited.
There was a problem hiding this comment.
Hmmm I'm curious why that is the case, I'll investigate
| @@ -125,6 +128,10 @@ def to_raw(self): | |||
| self._raw = Image.open(self._path) | |||
| return self._raw | |||
|
|
|||
| if self._tensor is not None: | |||
There was a problem hiding this comment.
This case was missing in the original function
| @@ -125,6 +128,10 @@ def to_raw(self): | |||
| self._raw = Image.open(self._path) | |||
| return self._raw | |||
|
|
|||
| if self._tensor is not None: | |||
| array = self._tensor.cpu().detach().numpy() | |||
| return Image.fromarray((255 - array * 255).astype(np.uint8)) | |||
There was a problem hiding this comment.
I invert the image colors with this (255 - array*255), else the image appears in negative.
|
|
||
| class AgentAudio(AgentType): | ||
| class AgentAudio(AgentType, str): |
There was a problem hiding this comment.
I add the str type so that gradio postprocessor can recognize the AgentAudio as a path to an audio.
aymeric-roucher
force-pushed
the
fix-gradio-tool-demos
branch
3 times, most recently
from
306a52d
to
5163ab6
Compare
aymeric-roucher
force-pushed
the
fix-gradio-tool-demos
branch
2 times, most recently
from
b9c774e
to
af90dab
Compare
aymeric-roucher
force-pushed
the
fix-gradio-tool-demos
branch
from
af90dab
to
c886bf9
Compare
| @@ -88,7 +88,8 @@ class AgentImage(AgentType, ImageType): | |||
| """ | |||
|
|
|||
| def __init__(self, value): | |||
| super().__init__(value) | |||
| AgentType.__init__(self, value) | |||
| ImageType.__init__(self) | |||
There was a problem hiding this comment.
Hmmm I'm curious why that is the case, I'll investigate
* Fix gradio tool demos
What does this PR do?
Resolves Gradio demos on Spaces for tools not working properly.