Guxtavv commited on
Commit
a809454
1 Parent(s): c1f84a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,5 +1,12 @@
1
  import gradio as gr
2
  from hf_olmo import AutoModelForCausalLM, AutoTokenizer # Importando do hf_olmo
 
 
 
 
 
 
 
3
 
4
  # Carregar o modelo e tokenizer
5
  model_name = "allenai/OLMo-7B"
 
1
  import gradio as gr
2
  from hf_olmo import AutoModelForCausalLM, AutoTokenizer # Importando do hf_olmo
3
+ import os
4
+
5
+ # Configurar PKG_CONFIG_PATH
6
+ os.environ['PKG_CONFIG_PATH'] = '/usr/local/lib/pkgconfig:/usr/lib/pkgconfig'
7
+
8
+ # Resto do código do app
9
+ from transformers import AutoTokenizer, AutoModelForCausalLM
10
 
11
  # Carregar o modelo e tokenizer
12
  model_name = "allenai/OLMo-7B"