Update app.py

app.py

@@ -111,7 +111,7 @@ if uploaded_file is not None and process_button:
             # Process the tensor through the model
             with st.spinner('Processing the tensor through the model...'):
                 with torch.no_grad():
-                    output = model.encode(tensor, use_ema=model.config.test_ema) 
+                    output = model(tensor) 
                     if isinstance(output, tuple):
                         output = output[0]
             output = output.squeeze(0)