Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 
 import torch
+import dspy
 from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStreamer, pipeline
 from threading import Thread
 
@@ -21,9 +22,9 @@ phi2 = pipeline(
     device_map="cpu"
 )
 # DSPy-based prompt generation
-from dspy_ai.agents import Agent
-from dspy_ai import spawn_processes
-from dspy_ai.utils import SentenceSplitter, SentimentAnalyzer, NamedEntityRecognizer
+from dspy.agents import Agent
+from dspy import spawn_processes
+from dspy.utils import SentenceSplitter, SentimentAnalyzer, NamedEntityRecognizer
 
 def dspy_generate_agent_prompts(prompt):
     """