Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,9 @@ import dspy
|
|
5 |
from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStreamer, pipeline
|
6 |
from threading import Thread
|
7 |
|
|
|
|
|
|
|
8 |
# The huggingface model id for Microsoft's phi-2 model
|
9 |
checkpoint = "microsoft/phi-2"
|
10 |
|
@@ -24,7 +27,6 @@ phi2 = pipeline(
|
|
24 |
# DSPy-based prompt generation
|
25 |
# from dspy import Agent
|
26 |
# from dspy import spawn_processes
|
27 |
-
from dspy.utils import SentenceSplitter, SentimentAnalyzer, NamedEntityRecognizer
|
28 |
|
29 |
def dspy_generate_agent_prompts(prompt):
|
30 |
"""
|
|
|
5 |
from transformers import AutoTokenizer, AutoModelForCausalLM, TextIteratorStreamer, pipeline
|
6 |
from threading import Thread
|
7 |
|
8 |
+
from sentence_splitter import SentenceSplitter # the following are made up imports (?), SentimentAnalyzer, NamedEntityRecognizer
|
9 |
+
|
10 |
+
|
11 |
# The huggingface model id for Microsoft's phi-2 model
|
12 |
checkpoint = "microsoft/phi-2"
|
13 |
|
|
|
27 |
# DSPy-based prompt generation
|
28 |
# from dspy import Agent
|
29 |
# from dspy import spawn_processes
|
|
|
30 |
|
31 |
def dspy_generate_agent_prompts(prompt):
|
32 |
"""
|