Monke64 commited on
Commit
3b31fa6
·
1 Parent(s): 3532ba1

Added image model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  from flask.Emotion_spotting_service import _Emotion_spotting_service
3
  from flask.Genre_spotting_service import _Genre_spotting_service
4
  from flask.Beat_tracking_service import _Beat_tracking_service
5
- #from diffusers import StableDiffusionPipeline
6
  import torch
7
 
8
  emo_list = []
@@ -41,7 +41,7 @@ if 'beat' not in st.session_state:
41
  emotion_service = load_emo_model()
42
  genre_service = load_genre_model()
43
  beat_service = load_beat_model()
44
- #image_service = load_image_model()
45
 
46
  st.title("Music2Image webpage")
47
  user_input = st.file_uploader("Upload your wav/mp3 files here", type=["wav","mp3"],key = "file_uploader")
 
2
  from flask.Emotion_spotting_service import _Emotion_spotting_service
3
  from flask.Genre_spotting_service import _Genre_spotting_service
4
  from flask.Beat_tracking_service import _Beat_tracking_service
5
+ from diffusers import StableDiffusionPipeline
6
  import torch
7
 
8
  emo_list = []
 
41
  emotion_service = load_emo_model()
42
  genre_service = load_genre_model()
43
  beat_service = load_beat_model()
44
+ image_service = load_image_model()
45
 
46
  st.title("Music2Image webpage")
47
  user_input = st.file_uploader("Upload your wav/mp3 files here", type=["wav","mp3"],key = "file_uploader")