Update app.py

app.py

@@ -141,12 +141,12 @@ def main():
         fig = plt.figure(figsize=(12,6))
         ax = [fig.add_subplot(121, axisbelow=True),fig.add_subplot(122)]
         ax[0].set_title('Monkeypox confirmed cases')
-        ax[0].plot(all_location[country]['total_cases'],lw=3,label='Infective')
+        ax[0].plot(all_location[country_code]['total_cases'],lw=3,label='Infective')
         ax[0].set_xlabel('Days')
         ax[0].set_ylabel('Number of cases')
         ax[0].legend()
         
-        scaler = all_location[country]['total_cases'].apply(lambda x : x/pop)
+        scaler = all_location[country_code]['total_cases'].apply(lambda x : x/pop)
         ax[1].set_title('Monkeypox confirmed cases compare with model')
         ax[1].plot(scaler,lw=3,label='Real Infective')
         ax[1].plot(i,lw=3,label='SIR model Infective')