saritha commited on
Commit
0248cb9
·
verified ·
1 Parent(s): 890db39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,7 +42,7 @@ class_names = ['BacterialBlights', 'Healthy', 'Mosaic', 'RedRot', 'Rust', 'Yello
42
 
43
  #Gemini Response
44
  def get_response_llm(predicted_label,knowledge_base):
45
- prompt = f"Your an helpful assistant who helps farmers know about the sugarcane leaf diseases , precaution, advise etc....Predicted disease label will is given to you '{predicted_label}' and also {knowledge_base} Provide detailed advice for managing this condition."
46
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
47
  model = genai.GenerativeModel("gemini-1.5-flash")
48
  response = model.generate_content([prompt])
@@ -161,7 +161,7 @@ def predict_disease(image):
161
  # detailed_response = knowledge_base[predicted_label]
162
  # else:
163
  # # Fallback to AI-generated response
164
-
165
  detailed_response = get_response_llm(predicted_label,knowledge_base)
166
 
167
  # Create a styled HTML output
 
42
 
43
  #Gemini Response
44
  def get_response_llm(predicted_label,knowledge_base):
45
+ prompt = f"Your an helpful assistant who helps farmers know about the sugarcane leaf diseases , precaution, advise etc....Predicted disease label will is given to you '{predicted_label}' and also {knowledge_base} Provide breif answer of advise for managing this condition."
46
  genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
47
  model = genai.GenerativeModel("gemini-1.5-flash")
48
  response = model.generate_content([prompt])
 
161
  # detailed_response = knowledge_base[predicted_label]
162
  # else:
163
  # # Fallback to AI-generated response
164
+ predicted_label = f'The predicted label is {predicted_label}'
165
  detailed_response = get_response_llm(predicted_label,knowledge_base)
166
 
167
  # Create a styled HTML output