Skip to content

Commit

Permalink
Summarization through Anthropic and OpenAI working
Browse files Browse the repository at this point in the history
Summarization through OpenAI and Anthropic's APIs are verified working. Next step is local API endpoints.
  • Loading branch information
rmusser01 committed May 5, 2024
1 parent e3e121a commit c2c7a3c
Show file tree
Hide file tree
Showing 3 changed files with 202 additions and 109 deletions.
6 changes: 5 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -34,9 +34,13 @@ options:
1. Setup python + packages
2. Setup ffmpeg
3. Run `python diarize.py <video_url>` or `python diarize.py <List_of_videos.txt>`
4. If you want summarization, add your API keys (if needed[is needed for now]) to the `config.txt` file, and then re-run the script, passing in the URL endpoint of the API you want to use.
4. If you want summarization, add your API keys (if needed[is needed for now]) to the `config.txt` file, and then re-run the script, passing in the name of the API [or URL endpoint - to be added] to the script.
* `python diarize.py https://www.youtube.com/watch?v=4nd1CDZP21s --api_name anthropic` - This will attempt to download the video, then upload the resulting json file to the anthropic API endpoint, referring to values set in the config file (API key and model) to request summarization.
- OpenAI:
- Anthropic:
* Opus: `claude-3-opus-20240229`
* Sonnet: `claude-3-sonnet-20240229`
* Haiku: `claude-3-haiku-20240307`
- Cohere:

### What's in the repo?
Expand Down
10 changes: 5 additions & 5 deletions config.txt
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
[API]
openai_api_key = your_openai_api_key
anthropic_api_key = your_anthropic_api_key
anthropic_model = claude-v1
openai_api_key = <openai_api_key_value>
openai_model = gpt-3.5-turbo
cohere_api_key = your_cohere_api_key
anthropic_api_key = <anthropic_api_Key_value>
anthropic_model = claude-3-sonnet-20240229
cohere_api_key = <your_cohere_api_key>
cohere_model = base

[Paths]
output_path = Results

[Processing]
processing_choice = cuda
processing_choice = cuda
Loading

0 comments on commit c2c7a3c

Please sign in to comment.