Update app.py

app.py

@@ -17,6 +17,9 @@ import PIL.Image
 import tensorflow as tf
 import piexif
 import piexif.helper
+import requests
+from io import BytesIO
+from PIL import Image
 
 TITLE = 'DeepDanbooru String'
 
@@ -68,8 +71,12 @@ def plaintext_to_html(text):
     text = "<p>" + "<br>\n".join([f"{html.escape(x)}" for x in text.split('\n')]) + "</p>"
     return text
 
-def predict(image: PIL.Image.Image, score_threshold: float,
+def predict(image_url:str, score_threshold: float,
             model: tf.keras.Model, labels: list[str]) -> dict[str, float]:
+    response = requests.get(image_url)
+
+    # 将图片内容转换为Pillow.Image对象
+    image = Image.open(BytesIO(response.content))
     rawimage = image
     _, height, width, _ = model.input_shape
     image = np.asarray(image)
@@ -140,7 +147,7 @@ def main():
     gr.Interface(
         func,
         [
-            gr.inputs.Image(type='pil', label='Input'),
+            gr.inputs.Textbox( label='Inputurl'),
             gr.inputs.Slider(0,
                              1,
                              step=args.score_slider_step,