SergioCabrera commited on
Commit
73a64e1
·
1 Parent(s): 85a2b11

Subida de la demo

Browse files
Files changed (2) hide show
  1. demo.py → app.py +2 -3
  2. requirements +3 -0
demo.py → app.py RENAMED
@@ -1,5 +1,5 @@
1
  # -*- coding: utf-8 -*-
2
- """Demo.ipynb
3
 
4
  Automatically generated by Colaboratory.
5
 
@@ -12,12 +12,11 @@ Original file is located at
12
 
13
  import tensorflow as tf
14
  from huggingface_hub import from_pretrained_keras
15
- import numpy as np
16
  import gradio as gr
17
 
18
  model = from_pretrained_keras("keras-io/supervised-contrastive-learning-cifar10")
19
 
20
- labels = ["avion", "pajaro", "coche", "gato", "perro", "caballo", "barco", "camion"]
21
 
22
  def clasificadorImages(test_imagen):
23
  image = tf.constant(test_imagen)
 
1
  # -*- coding: utf-8 -*-
2
+ """app.ipynb
3
 
4
  Automatically generated by Colaboratory.
5
 
 
12
 
13
  import tensorflow as tf
14
  from huggingface_hub import from_pretrained_keras
 
15
  import gradio as gr
16
 
17
  model = from_pretrained_keras("keras-io/supervised-contrastive-learning-cifar10")
18
 
19
+ labels = ["avion", "pajaro", "coche", "gato", "ciervo","perro", "caballo", "mono", "barco", "camion"]
20
 
21
  def clasificadorImages(test_imagen):
22
  image = tf.constant(test_imagen)
requirements ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ tensorflow
2
+ huggingface_hub
3
+ gradio