Tonic commited on
Commit
55fee33
1 Parent(s): 1b5413e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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
  """