Views
No views yet
1pip install transformers torch tensorflow opencv-python flask
2
3from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
5tokenizer = AutoTokenizer.from_pretrained("agri_expert/agrigpt")
6model = AutoModelForSeq2SeqLM.from_pretrained("agri_expert/agrigpt")
7
8def get_response(query):
9 inputs = tokenizer(query, return_tensors="pt")
10 outputs = model.generate(**inputs)
11 response = tokenizer.decode(outputs[0], skip_special_tokens=True)
12 return response
13
14 import tensorflow as tf
15
16# Load the model
17model = tf.keras.models.load_model("crop_monitoring/model.h5")
18
19# Predict crop health
20def predict_crop_health(image_path):
21 img = tf.keras.preprocessing.image.load_img(image_path, target_size=(224, 224))
22 img_array = tf.keras.preprocessing.image.img_to_array(img)
23 img_array = tf.expand_dims(img_array, 0) / 255.0
24 predictions = model.predict(img_array)
25 label = labels[np.argmax(predictions)]
26 return label
27
28 Requirements
29Python 3.8+
30Libraries: transformers, torch, tensorflow, opencv-python, flask
31