amasood commited on
Commit
dccf025
·
verified ·
1 Parent(s): 8198ae8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ 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
- client = Groq(api_key=os.environ.get(GROQ_API_KEY))
20
 
21
  # Initialize session state for disease details
22
  if "disease_name" not in st.session_state:
 
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
+ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
20
 
21
  # Initialize session state for disease details
22
  if "disease_name" not in st.session_state: