paloma99 commited on
Commit
32c1b72
1 Parent(s): 2136930

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -36,6 +36,9 @@ import shutil
36
 
37
 
38
 
 
 
 
39
  # Cell 1: Image Classification Model
40
  image_pipeline = pipeline(task="image-classification", model="guillen/vit-basura-test1")
41
 
@@ -47,7 +50,7 @@ image_gradio_app = gr.Interface(
47
  fn=predict_image,
48
  inputs=gr.Image(label="Image", sources=['upload', 'webcam'], type="pil"),
49
  outputs=[gr.Label(label="Result")],
50
- title="Green Greta",
51
  theme=theme
52
  )
53
 
 
36
 
37
 
38
 
39
+ custom_title = "<span style='color: rgb(18, 13, 5);'>Green Greta</span>"
40
+
41
+
42
  # Cell 1: Image Classification Model
43
  image_pipeline = pipeline(task="image-classification", model="guillen/vit-basura-test1")
44
 
 
50
  fn=predict_image,
51
  inputs=gr.Image(label="Image", sources=['upload', 'webcam'], type="pil"),
52
  outputs=[gr.Label(label="Result")],
53
+ title=custom_title,
54
  theme=theme
55
  )
56