Quintino Fernandes commited on
Commit
fc4aec6
·
1 Parent(s): a2682b3

Bug fixing

Browse files
Files changed (3) hide show
  1. Dockerfile +6 -1
  2. app_antiga.py +2 -2
  3. database/query.py +1 -1
Dockerfile CHANGED
@@ -19,9 +19,14 @@ COPY --chown=user ./requirements.txt requirements.txt
19
  RUN pip install --no-cache-dir --upgrade pip && \
20
  pip install --no-cache-dir -r requirements.txt
21
 
22
- # Download Spacy model
23
  RUN python -m spacy download pt_core_news_md
24
 
 
 
 
 
 
25
  # Copy application code
26
  COPY --chown=user . /app
27
 
 
19
  RUN pip install --no-cache-dir --upgrade pip && \
20
  pip install --no-cache-dir -r requirements.txt
21
 
22
+ # Pre-download Spacy model
23
  RUN python -m spacy download pt_core_news_md
24
 
25
+ # Pre-download T5 models
26
+ RUN python -c "from transformers import T5Tokenizer, T5ForConditionalGeneration; \
27
+ T5Tokenizer.from_pretrained('unicamp-dl/ptt5-base-portuguese-vocab'); \
28
+ T5ForConditionalGeneration.from_pretrained('recogna-nlp/ptt5-base-summ')"
29
+
30
  # Copy application code
31
  COPY --chown=user . /app
32
 
app_antiga.py CHANGED
@@ -29,8 +29,8 @@ app.add_middleware(
29
  )
30
 
31
  #Database connection setup
32
- url = "https://daxquaudqidyeirypexa.supabase.co"
33
- key = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImRheHF1YXVkcWlkeWVpcnlwZXhhIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NDQzOTIzNzcsImV4cCI6MjA1OTk2ODM3N30.3qB-GfiCoqXEpbNfqV3iHiqOLr8Ex9nPVr6p9De5Hdc"
34
  opts = ClientOptions().replace(schema="articles")
35
  supabase = create_client(url, key, options=opts)
36
 
 
29
  )
30
 
31
  #Database connection setup
32
+ url = ""
33
+ key = ""
34
  opts = ClientOptions().replace(schema="articles")
35
  supabase = create_client(url, key, options=opts)
36
 
database/query.py CHANGED
@@ -48,7 +48,7 @@ class DatabaseService:
48
  )
49
 
50
  # Format results with metadata
51
- formatted_results = []-
52
  for article_id, distance in results:
53
  metadata = self.articles.fetch(ids=[article_id])[0]["metadata"]
54
  formatted_results.append({
 
48
  )
49
 
50
  # Format results with metadata
51
+ formatted_results = []
52
  for article_id, distance in results:
53
  metadata = self.articles.fetch(ids=[article_id])[0]["metadata"]
54
  formatted_results.append({