Update app.py

app.py

@@ -3,34 +3,33 @@ import openai
 import os
 
 
-def summarize(prompt):
+def classify(prompt):
     try:
-        augmented_prompt = f"summarize this text: {prompt}"
+        augmented_prompt = f"{prompt} \n\n###\n\n"
         st.session_state["summary"] = openai.Completion.create(
-            model= "text-davinci-003",
+            model="ada:ft-personal:prt-model-2023-02-10-07-36-17",#"text-davinci-003",
             prompt=augmented_prompt,
             temperature=.3,
-            max_tokens=150,
+            max_tokens=1,
         )["choices"][0]["text"]
     except:
         st.write('There was an error =(')
 
-# openai.api_key = os.getenv('OPENAI_KEY')
 try:
     if "summary" not in st.session_state:
         st.session_state["summary"] = ""
     openai_api_key = st.text_input("Enter Your OpenAI API key: ", key="key_input",type="password")
     openai.api_key  = openai_api_key
-    st.title("Text Summarizer")
+    st.title("Text Classification")
 
-    input_text = st.text_area(label="Enter full text:", value="", height=250)
+    input_text = st.text_area(label="Enter full text:", value="", height=300)
 
     st.button(
         "Submit",
-        on_click=summarize,
+        on_click=classify,
         kwargs={"prompt": input_text},
     )
 
-    output_text = st.text_area(label="Summarized text:", value=st.session_state["summary"], height=250)
+    output_text = st.text_area(label="Sentiment of text:", value=st.session_state["summary"], height=5)
 except:
   st.write('An error occured, Please contact the developer')