Update app.py

app.py

@@ -11,12 +11,14 @@ openai.api_key = os.getenv('OPENAI_API_KEY')
 
 # Function to classify email
 def classify_email(email_text):
-    response = openai.Completion.create(
-        model='gpt-3.5-turbo-instruct',
-        prompt=f'Classify the following email:\n\n{email_text}\n\nCategories: Spam, Work, Personal, Promotion, Other',
-        max_tokens=50
+    response = openai.ChatCompletion.create(
+        model='gpt-3.5-turbo',
+        messages=[
+            {"role": "system", "content": "You are an email classifier."},
+            {"role": "user", "content": f"Classify the following email:\n\n{email_text}\n\nCategories: Spam, Work, Personal, Promotion, Other"}
+        ]
     )
-    return response.choices[0].text.strip()
+    return response['choices'][0]['message']['content'].strip()
 
 # Streamlit interface
 st.title('Email Classifier')