A little bit of weight loss is normal after any illness in small babies, mostly this should recover with a normal diet, you can add some high calorie foods like ghee, grated dry fruits, paneer, सोया etc.
Consult your pediatrician if the weight gain does not happen
Health Tips
For more tips about child health you can check out my you tube channel as well... https://www.youtube.com/channel/UCHhuutqc0z5CrHePURBS_Bg