Msaqibsharif commited on
Commit
cf85e96
·
verified ·
1 Parent(s): ad4dc06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,8 +12,8 @@ processor = AutoImageProcessor.from_pretrained(model_name)
12
  model = AutoModelForImageClassification.from_pretrained(model_name)
13
 
14
  # Set your OpenAI API key from environment variable
15
- openai_api_key = os.getenv("OPENAI_API_KEY")
16
- openai.api_key = openai_api_key
17
 
18
  # Dropdown Options
19
  car_companies = ["Select", "Toyota", "Honda", "Ford", "BMW", "Mercedes", "Audi", "Hyundai", "Kia", "Nissan"]
@@ -41,7 +41,7 @@ def estimate_repair_cost(damage_type, company, model, year, country):
41
  )
42
 
43
  try:
44
- response = openai.ChatCompletion.create(
45
  model="gpt-4o-mini",
46
  messages=[
47
  {"role": "system", "content": "You are an expert in car repair cost estimation."},
 
12
  model = AutoModelForImageClassification.from_pretrained(model_name)
13
 
14
  # Set your OpenAI API key from environment variable
15
+ openai_api_key = os.getenv("OpenAI4oMini")
16
+ client = openai.OpenAI(api_key=openai_api_key) # Required for API calls
17
 
18
  # Dropdown Options
19
  car_companies = ["Select", "Toyota", "Honda", "Ford", "BMW", "Mercedes", "Audi", "Hyundai", "Kia", "Nissan"]
 
41
  )
42
 
43
  try:
44
+ response = client.ChatCompletion.create(
45
  model="gpt-4o-mini",
46
  messages=[
47
  {"role": "system", "content": "You are an expert in car repair cost estimation."},