Update app.py

app.py

@@ -26,16 +26,23 @@ def load_pipeline(task, model):
 
 # Function to predict using Hugging Face models and OpenAI models
 def predict(task, model, text):
-    if model in ["ChatGPT", "GPT-4"]:
-        response = openai.ChatCompletion.create(
-            model="gpt-4" if model == "GPT-4" else "gpt-3.5-turbo",
-            messages=[{"role": "user", "content": text}]
-        )
-        results = [{"label": response.choices[0].message['content'].strip()}]
-    else:
+    try:
         selected_pipeline = load_pipeline(task, model)
-        results = selected_pipeline(text)
-    return results
+        if model in ["ChatGPT", "GPT4"]:
+            # OpenAI API request
+            response = openai.ChatCompletion.create(
+                model=MODELS[model],
+                messages=[{"role": "user", "content": text}]
+            )
+            return response['choices'][0]['message']['content']
+        else:
+            # Hugging Face pipeline
+            results = selected_pipeline(text)
+            return results
+    except Exception as e:
+        print(f"Error in prediction: {e}")
+        return {"error": str(e)}
+
 
 # Function to benchmark Hugging Face models and OpenAI models
 def benchmark(task, model, file):