Jfink09 commited on
Commit
cbe9e87
·
1 Parent(s): 4142d44

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -64,9 +64,9 @@ def predict(img) -> Tuple[Dict, float]:
64
  ### 4. Gradio app ###
65
 
66
  # Create title, description and article strings
67
- title = "DeepFundus 👀"
68
- description = "A ResNet50 feature extractor computer vision model to classify funduscopic images."
69
- article = "Created with the help from [09. PyTorch Model Deployment](https://www.learnpytorch.io/09_pytorch_model_deployment/)."
70
 
71
  # Create examples list from "examples/" directory
72
  example_list = [["examples/" + example] for example in os.listdir("examples")]
@@ -80,7 +80,10 @@ demo = gr.Interface(fn=predict, # mapping function from input to output
80
  examples=example_list,
81
  title=title,
82
  description=description,
83
- article=article)
 
 
 
84
 
85
  # Launch the demo!
86
- demo.launch()
 
64
  ### 4. Gradio app ###
65
 
66
  # Create title, description and article strings
67
+ #title = "DeepFundus 👀"
68
+ #description = "A ResNet50 feature extractor computer vision model to classify funduscopic images."
69
+ #article = "Created with the help from [09. PyTorch Model Deployment](https://www.learnpytorch.io/09_pytorch_model_deployment/)."
70
 
71
  # Create examples list from "examples/" directory
72
  example_list = [["examples/" + example] for example in os.listdir("examples")]
 
80
  examples=example_list,
81
  title=title,
82
  description=description,
83
+ article=article,
84
+ css="""
85
+ .gradio-container {background-color: #000029}
86
+ """)
87
 
88
  # Launch the demo!
89
+ demo.launch()