Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import os
 import torch
 
-from model import create_effnetv2_model
+from model import create_effnetv2_m_model
 from timeit import default_timer as timer
 from typing import Tuple, Dict
 
@@ -14,7 +14,7 @@ with open("class_names.txt", "r") as f:  # reading them in from class_names.txt
 ### 2. Model and transforms preparation ###
 
 # Create model
-effnetv2, effnetv2_transforms = create_effnetv2_model(
+effnetv2, effnetv2_transforms = create_effnetv2_m_model(
     num_classes=101,  # could also use len(class_names)
 )