From ce024dfebcf677c00160d990ecc76aba4ffc03a8 Mon Sep 17 00:00:00 2001
From: Akash Choudhuri <63670764+Soothysay@users.noreply.github.com>
Date: Thu, 1 Jun 2023 15:08:58 -0700
Subject: [PATCH] Update README.md

Added a method call to reduce import error
---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index cfc3ced..3720e37 100644
--- a/README.md
+++ b/README.md
@@ -197,7 +197,7 @@ The following example shows how to profile AlexNet using the Flops Profiler.
 ```python
 import torchvision.models as models
 import torch
-from flops_profiler import get_model_profile
+from flops_profiler.profiler import get_model_profile
 
 with torch.cuda.device(0):
     model = models.alexnet()
@@ -234,7 +234,7 @@ The `FlopsProfiler`class provides the following methods:
 Below is an example of this usage in a typical training workflow.
 
 ```python
-from flops_profiler import FlopsProfiler
+from flops_profiler.profiler import FlopsProfiler
 
 model = Model()
 prof = FlopsProfiler(model, ds_engine if ds_engine else None)