cambios

Files changed (16) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model, transform = create_effnetb2_model(
 model.load_state_dict(
     torch.load(
-        f="model_v5.pth",
         map_location=torch.device("cpu")
     )
 )
@@ -46,7 +46,7 @@ example_list = [["examples/" + example] for example in os.listdir("examples")]
 # Gradio interface
 title = "Modelo de Clasificación de Imágenes Finetuneado (Chatbots)"
 description = "Clasifica el tipo de clima de una imagen"
-article = "Código en el que está basado [GitHub](https://github.com/georgescutelnicu/Weather-Image-Classification)."
 demo = gr.Interface(fn=predict,
                     inputs=gr.Image(type="pil"),

 model.load_state_dict(
     torch.load(
+        f="model_v3.pth",
         map_location=torch.device("cpu")
     )
 )
 # Gradio interface
 title = "Modelo de Clasificación de Imágenes Finetuneado (Chatbots)"
 description = "Clasifica el tipo de clima de una imagen"
+article = "Código en el que está basado  [GitHub](https://github.com/georgescutelnicu/Weather-Image-Classification)."
 demo = gr.Interface(fn=predict,
                     inputs=gr.Image(type="pil"),