Jeysshon commited on
Commit
4ced22e
1 Parent(s): 55316a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -3,10 +3,10 @@ from fastai.vision.all import *
3
  import gradio as gr
4
 
5
  # Cargar los modelos
6
- learn_emotion = load_learner('/content/drive/MyDrive/carpeta de PIB/emotions_javier_almeida.pkl')
7
  learn_emotion_labels = learn_emotion.dls.vocab
8
 
9
- learn_sentiment = load_learner('/content/drive/MyDrive/carpeta de PIB/sentiment_javier_almeida.pkl')
10
  learn_sentiment_labels = learn_sentiment.dls.vocab
11
 
12
  # Función de predicción
@@ -29,8 +29,14 @@ description = (
29
  article = "Esta herramienta proporciona una forma rápida de analizar emociones y sentimientos en imágenes."
30
 
31
  examples = [
32
- '/content/drive/MyDrive/carpeta de PIB/angry1.png',
33
- '/content/drive/MyDrive/carpeta de PIB/neutral1.jpg'
 
 
 
 
 
 
34
  ]
35
 
36
  iface = gr.Interface(
 
3
  import gradio as gr
4
 
5
  # Cargar los modelos
6
+ learn_emotion = load_learner('emotions_jey.pkl')
7
  learn_emotion_labels = learn_emotion.dls.vocab
8
 
9
+ learn_sentiment = load_learner('sentiment_jey.pkl')
10
  learn_sentiment_labels = learn_sentiment.dls.vocab
11
 
12
  # Función de predicción
 
29
  article = "Esta herramienta proporciona una forma rápida de analizar emociones y sentimientos en imágenes."
30
 
31
  examples = [
32
+ 'PrivateTest_10131363.jpg',
33
+ 'angry1.png',
34
+ 'angry2.jpg',
35
+ 'happy1.jpg',
36
+ 'happy2.jpg',
37
+ 'neutral1.jpg',
38
+ 'neutral2.jpg'
39
+
40
  ]
41
 
42
  iface = gr.Interface(