Check prime number i am getting error

[Yvardhan (ken kaneki)]

#include
using namespace std;
int main() {
int num,i,count=0;
cin>>num;
for(i=2;i<num;i++)
if(num%i==0)
count=count+1;

if(count!=1)
cout<<"prime";
else
cout<<"not prime";
return 0;

}