acecalisto3 commited on
Commit
6661be0
·
verified ·
1 Parent(s): c8f2a49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -3,10 +3,13 @@ from transformers import pipeline
3
  from sentence_transformers import SentenceTransformer, util
4
  import os
5
  import requests
6
- import logging
7
 
8
- # Configure logging for detailed insights
9
- logging.basicConfig(level=logging.INFO)
 
 
 
 
10
 
11
  # Constants for enhanced organization
12
  GITHUB_API_BASE_URL = "https://api.github.com/repos"
 
3
  from sentence_transformers import SentenceTransformer, util
4
  import os
5
  import requests
 
6
 
7
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
8
+
9
+ model_name = "enricoros/big-agi"
10
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
11
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
12
+
13
 
14
  # Constants for enhanced organization
15
  GITHUB_API_BASE_URL = "https://api.github.com/repos"