kritsg commited on
Commit
b590d13
·
1 Parent(s): 4790b80

starting on upload image functionality

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -10,6 +10,7 @@ from os.path import exists, dirname
10
  import sys
11
  import json
12
  import flask
 
13
 
14
  parent_dir = dirname(os.path.abspath(os.getcwd()))
15
  sys.path.append(parent_dir)
@@ -31,11 +32,15 @@ def get_image_data(image_name):
31
  def segmentation_generation(image_name, c_width, n_top, n_gif_imgs):
32
  print("Inputs Received:", image_name, c_width, n_top, n_gif_imgs)
33
 
 
 
 
 
34
  # cred_width = c_width
35
  # n_top_segs = n_top
36
  # n_gif_images = n_gif_imgs
37
  image, model_and_data = get_image_data(image_name)
38
- print("model_and_data", model_and_data)
39
  # Unpack datax
40
  xtest = model_and_data["xtest"]
41
  ytest = model_and_data["ytest"]
@@ -86,7 +91,8 @@ if __name__ == "__main__":
86
  # gradio's image inputs look like this: <PIL.Image.Image image mode=RGB size=305x266 at 0x7F3D01C91FA0>
87
  # need to learn how to handle image inputs, or deal with file inputs or just file path strings
88
  inp = gr.inputs.Textbox(lines=1, placeholder="Select an example from below", default="", label="Input Image Path", optional=False)
89
- out = [gr.outputs.HTML(label="Output GIF"), 'text']
 
90
 
91
  iface = gr.Interface(
92
  segmentation_generation,
 
10
  import sys
11
  import json
12
  import flask
13
+ from PIL import Image
14
 
15
  parent_dir = dirname(os.path.abspath(os.getcwd()))
16
  sys.path.append(parent_dir)
 
32
  def segmentation_generation(image_name, c_width, n_top, n_gif_imgs):
33
  print("Inputs Received:", image_name, c_width, n_top, n_gif_imgs)
34
 
35
+ print("imagename", image_name.filename)
36
+
37
+ return "yeehaw"
38
+
39
  # cred_width = c_width
40
  # n_top_segs = n_top
41
  # n_gif_images = n_gif_imgs
42
  image, model_and_data = get_image_data(image_name)
43
+ # print("model_and_data", model_and_data)
44
  # Unpack datax
45
  xtest = model_and_data["xtest"]
46
  ytest = model_and_data["ytest"]
 
91
  # gradio's image inputs look like this: <PIL.Image.Image image mode=RGB size=305x266 at 0x7F3D01C91FA0>
92
  # need to learn how to handle image inputs, or deal with file inputs or just file path strings
93
  inp = gr.inputs.Textbox(lines=1, placeholder="Select an example from below", default="", label="Input Image Path", optional=False)
94
+ out = 'text'
95
+ # out = [gr.outputs.HTML(label="Output GIF"), gr.outputs.Textbox(label="Prediction")]
96
 
97
  iface = gr.Interface(
98
  segmentation_generation,