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}],
|