Hugging Face's logo

Spaces:
Loren
/
search_articles
Sleeping

App Files Community
Loren commited on
Commit
4b4c6a7
·
verified ·
1 Parent(s): 7389064

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
  import time
6
  from transformers import AutoModelForCausalLM, AutoTokenizer
7
  from templates.prompt_mistral_rag import RAG_PROMPT_TEMPLATE
8
-
9
 
10
  # URL du backend Hugging Face Space
11
  API_URL = os.getenv('API_URL')
 
5
  import time
6
  from transformers import AutoModelForCausalLM, AutoTokenizer
7
  from templates.prompt_mistral_rag import RAG_PROMPT_TEMPLATE
8
+ import torch
9
 
10
  # URL du backend Hugging Face Space
11
  API_URL = os.getenv('API_URL')