Update app.py
Browse files
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,
|
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}],
|