Spaces:

garyd1
/

Excel_chatbot

Running

garyd1 commited on Feb 20

Commit

100cf1c

verified ·

1 Parent(s): 9175756

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -75,9 +75,10 @@ elif model_choice == "Claude 3 Haiku":
     def ask_claude(query):
         response = client.messages.create(
             model="claude-3-haiku",
             messages=[{"role": "user", "content": query}]
         )
-        return response.content
 else:
     client = openai.OpenAI()

     def ask_claude(query):
         response = client.messages.create(
             model="claude-3-haiku",
+            max_tokens=512,
             messages=[{"role": "user", "content": query}]
         )
+        return response.content[0]["text"]
 else:
     client = openai.OpenAI()