Update README.md

README.md

@@ -32,18 +32,12 @@ Users (both direct and downstream) should be made aware of the risks, biases and
 ## How to Get Started with the Model
 
 Use the code below to get started with the model.
-from tensorflow.keras.models import load_model
 
-# Load the pre-trained model
-model = load_model('path_to_model.h5')
+import torch 
+from torchvision import models 
+import pickle 
 
-# Example input data (image and numerical features)
-image_input = ...  # Preprocessed image data of shape (150, 150, 3)
-numerical_input = ...  # Numerical features (RSI, MACD, etc.)
-
-# Make a prediction
-prediction = model.predict([image_input, numerical_input])
-print("Predicted Stock Movement:", prediction)
+# Load the model