Update app.py

app.py

@@ -78,9 +78,9 @@ def load_pipeline():
     # model.load_state_dict(torch.load(model_file))
     model = load_model(model, model_file)
     # model = pipeline('image-classification', model=model_id, device='cpu') 
-    model.eval()
-    
-    return model.to('cpu')
+    # model.eval()
+    print(model)
+    return model#.to('cpu')
 
 # Streamlit app UI template
 st.title("Signature Forgery Detection")