ardavey commited on
Commit
11666df
1 Parent(s): 39a7b7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -1
app.py CHANGED
@@ -13,7 +13,27 @@ model.eval()
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  model.to(device)
15
 
16
- # Streamlit app
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  st.title("Skin Condition Identifier")
18
  st.write("Upload an image and provide a text prompt to identify the skin condition.")
19
 
 
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
  model.to(device)
15
 
16
+ # Add Hugging Face logo at the top
17
+ st.markdown(
18
+ """
19
+ <style>
20
+ .huggingface-logo {
21
+ display: flex;
22
+ justify-content: center;
23
+ margin-bottom: 20px;
24
+ }
25
+ .huggingface-logo img {
26
+ width: 150px;
27
+ }
28
+ </style>
29
+ <div class="huggingface-logo">
30
+ <img src="https://huggingface.co/front/assets/huggingface_logo-noborder.svg" alt="Hugging Face Logo">
31
+ </div>
32
+ """,
33
+ unsafe_allow_html=True,
34
+ )
35
+
36
+ # Streamlit app title and instructions
37
  st.title("Skin Condition Identifier")
38
  st.write("Upload an image and provide a text prompt to identify the skin condition.")
39