sergioska commited on
Commit
4bfcb8d
·
1 Parent(s): 525d1e9

fix file name

Browse files
Files changed (1) hide show
  1. app.py +17 -6
app.py CHANGED
@@ -3,14 +3,25 @@ import streamlit as st
3
  from transformers import pipeline
4
  from tempfile import NamedTemporaryFile
5
 
6
- pipe = pipeline("automatic-speech-recognition", model="openai/whisper-large-v3")
 
7
 
8
- st.title('Upload an audio file :blue[cool] :sunglasses:')
9
 
10
- uploaded_file = st.file_uploader("Choose a file")
11
- if uploaded_file is not None:
12
  with NamedTemporaryFile(suffix="wav") as temp:
13
- temp.write(uploaded_file.getvalue())
14
  temp.seek(0)
15
- result = pipe(temp.name)
 
 
 
 
 
 
 
 
 
 
16
  st.write(result)
 
3
  from transformers import pipeline
4
  from tempfile import NamedTemporaryFile
5
 
6
+ audiopipe = pipeline("automatic-speech-recognition", model="openai/whisper-large-v3")
7
+ imagepipe = pipeline("image-classification", model="Kaludi/food-category-classification-v2.0")
8
 
9
+ st.title('Upload an audio file for speech recognition')
10
 
11
+ uploaded_audio_file = st.file_uploader("Choose a file")
12
+ if uploaded_audio_file is not None:
13
  with NamedTemporaryFile(suffix="wav") as temp:
14
+ temp.write(uploaded_audio_file.getvalue())
15
  temp.seek(0)
16
+ result = audiopipe(temp.name)
17
+ st.write(result)
18
+
19
+ st.title('Upload an image file to classification (food)')
20
+
21
+ uploaded_image_file = st.file_uploader("Choose a file")
22
+ if uploaded_image_file is not None:
23
+ with NamedTemporaryFile() as temp:
24
+ temp.write(uploaded_image_file.getvalue())
25
+ temp.seek(0)
26
+ result = imagepipe(temp.name)
27
  st.write(result)