OmkarGhugarkar commited on
Commit
9265cfc
1 Parent(s): 32532af

1Word Sorted

Browse files
Files changed (2) hide show
  1. DescriptorPrompt.xlsx +0 -0
  2. utils.py +2 -1
DescriptorPrompt.xlsx CHANGED
Binary files a/DescriptorPrompt.xlsx and b/DescriptorPrompt.xlsx differ
 
utils.py CHANGED
@@ -22,6 +22,7 @@ pipe = pipeline("text-classification", model="mrm8488/distilroberta-finetuned-fi
22
  def call(prompt, text):
23
  client = Groq(api_key=os.getenv("key"),)
24
 
 
25
  chat_completion = client.chat.completions.create(
26
  messages=[
27
  {
@@ -82,7 +83,7 @@ def summary(input_json):
82
  prompt = "1 word Financial SEO tag for this news article"
83
  output['tag'] = call(prompt, output['mobile'])
84
 
85
- prompt = "Headline for this News Article"
86
  output['headline'] = call(prompt, output['web'])
87
 
88
  utc_now = datetime.datetime.utcnow()
 
22
  def call(prompt, text):
23
  client = Groq(api_key=os.getenv("key"),)
24
 
25
+ prompt = prompt + " Start Directly with the answer, don't give explanation of what you are doing"
26
  chat_completion = client.chat.completions.create(
27
  messages=[
28
  {
 
83
  prompt = "1 word Financial SEO tag for this news article"
84
  output['tag'] = call(prompt, output['mobile'])
85
 
86
+ prompt = "Give a single headline for this News Article"
87
  output['headline'] = call(prompt, output['web'])
88
 
89
  utc_now = datetime.datetime.utcnow()