amasood commited on
Commit
b638e4f
·
verified ·
1 Parent(s): 39d7e3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,7 +16,8 @@ model = AutoModelForImageClassification.from_pretrained(MODEL_NAME).to(DEVICE)
16
  processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
17
 
18
  # Set up the Groq API key (replace with your actual key or use an environment variable)
19
- GROQ_API_KEY = os.getenv("GROQ_API_KEY", seckey)
 
20
  client = Groq(api_key=GROQ_API_KEY)
21
 
22
  # Initialize session state for disease_name
@@ -39,7 +40,7 @@ def predict_skin_disease(image):
39
 
40
  # Function to get disease details from Groq API
41
  def get_disease_info(disease_name):
42
- prompt = f"Provide a detailed explanation about the skin disease '{disease_name}', including causes, symptoms, and treatment options."
43
 
44
  chat_completion = client.chat.completions.create(
45
  messages=[{"role": "user", "content": prompt}],
 
16
  processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
17
 
18
  # Set up the Groq API key (replace with your actual key or use an environment variable)
19
+ #GROQ_API_KEY = os.getenv("GROQ_API_KEY", seckey)
20
+ GROQ_API_KEY = seckey
21
  client = Groq(api_key=GROQ_API_KEY)
22
 
23
  # Initialize session state for disease_name
 
40
 
41
  # Function to get disease details from Groq API
42
  def get_disease_info(disease_name):
43
+ prompt = f"Provide a detailed explanation about the skin disease '{disease_name}', including description of disease, causes, precausions, risk and treatment options."
44
 
45
  chat_completion = client.chat.completions.create(
46
  messages=[{"role": "user", "content": prompt}],