saritha commited on
Commit
9e142cf
·
verified ·
1 Parent(s): af2c944

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ class_names = ['BacterialBlights', 'Healthy', 'Mosaic', 'RedRot', 'Rust', 'Yello
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="YOUR_API_KEY")
47
  model = genai.GenerativeModel("gemini-1.5-flash")
48
  response = model.generate_content([prompt])
49
  return response.text
 
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])
49
  return response.text