Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import argparse
|
2 |
import os
|
3 |
-
|
4 |
-
|
5 |
import gradio as gr
|
6 |
import huggingface_hub
|
7 |
import numpy as np
|
@@ -175,7 +175,7 @@ class Predictor:
|
|
175 |
):
|
176 |
self.load_model(model_repo)
|
177 |
|
178 |
-
image = self.prepare_image(
|
179 |
|
180 |
input_name = self.model.get_inputs()[0].name
|
181 |
label_name = self.model.get_outputs()[0].name
|
|
|
1 |
import argparse
|
2 |
import os
|
3 |
+
|
4 |
+
|
5 |
import gradio as gr
|
6 |
import huggingface_hub
|
7 |
import numpy as np
|
|
|
175 |
):
|
176 |
self.load_model(model_repo)
|
177 |
|
178 |
+
image = self.prepare_image(image)
|
179 |
|
180 |
input_name = self.model.get_inputs()[0].name
|
181 |
label_name = self.model.get_outputs()[0].name
|