Spaces:
Running
Running
Commit
·
263de18
1
Parent(s):
e0d81d7
🚫 Fix Wrapper
Browse files
App.py
CHANGED
|
@@ -307,10 +307,8 @@ with App.Blocks(
|
|
| 307 |
InputVideo = App.Video(
|
| 308 |
label='Input Video', sources=['upload'], height=300
|
| 309 |
)
|
| 310 |
-
|
| 311 |
-
|
| 312 |
-
|
| 313 |
-
ModelNames = GetModelNames()
|
| 314 |
InputModel = App.Dropdown(
|
| 315 |
choices=ModelNames,
|
| 316 |
label='Select Model',
|
|
@@ -425,20 +423,8 @@ with App.Blocks(
|
|
| 425 |
outputs=[InputThreshold, InputMinPercentage, InputMaxRectangles, InputPadding, InputSegmentRows, InputSegmentColumns, InputFullFrameInterval],
|
| 426 |
)
|
| 427 |
|
| 428 |
-
def ProcessWrapper(InputVideo, InputModel, InputUseRegions, InputThreshold, InputMinPercentage, InputMaxRectangles, InputPadding, Progress=App.Progress()):
|
| 429 |
-
return Upscaler().Process(
|
| 430 |
-
InputVideo,
|
| 431 |
-
InputModel,
|
| 432 |
-
InputUseRegions,
|
| 433 |
-
InputThreshold,
|
| 434 |
-
InputMinPercentage,
|
| 435 |
-
InputMaxRectangles,
|
| 436 |
-
InputPadding,
|
| 437 |
-
Progress
|
| 438 |
-
)
|
| 439 |
-
|
| 440 |
SubmitButton.click(
|
| 441 |
-
fn=
|
| 442 |
inputs=[
|
| 443 |
InputVideo,
|
| 444 |
InputModel,
|
|
|
|
| 307 |
InputVideo = App.Video(
|
| 308 |
label='Input Video', sources=['upload'], height=300
|
| 309 |
)
|
| 310 |
+
ModelList = Upscaler().ListModels()
|
| 311 |
+
ModelNames = [Path(Model).stem for Model in ModelList]
|
|
|
|
|
|
|
| 312 |
InputModel = App.Dropdown(
|
| 313 |
choices=ModelNames,
|
| 314 |
label='Select Model',
|
|
|
|
| 423 |
outputs=[InputThreshold, InputMinPercentage, InputMaxRectangles, InputPadding, InputSegmentRows, InputSegmentColumns, InputFullFrameInterval],
|
| 424 |
)
|
| 425 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 426 |
SubmitButton.click(
|
| 427 |
+
fn=Upscaler().Process,
|
| 428 |
inputs=[
|
| 429 |
InputVideo,
|
| 430 |
InputModel,
|