Bob-Potato commited on
Commit
d7226b7
·
verified ·
1 Parent(s): dc7acf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,8 +1,14 @@
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
  import torch
 
4
 
5
- # Model public, mic și gratuit
 
 
 
 
 
6
  MODEL_NAME = "google/flan-t5-small"
7
 
8
  # Încarcă model și tokenizer
@@ -28,7 +34,7 @@ iface = gr.Interface(
28
  inputs=gr.Textbox(lines=2, placeholder="Întreabă ceva..."),
29
  outputs="text",
30
  title="Chatbot simplu",
31
- description="Chatbot minimal bazat pe Flan-T5-small (fără date pre-trained locale)."
32
  )
33
 
34
  iface.launch(server_name="0.0.0.0", server_port=7860)
 
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
  import torch
4
+ import os
5
 
6
+ # Folder cache cu permisiune
7
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache"
8
+ os.environ["HF_HOME"] = "/tmp/.cache"
9
+ os.environ["HF_DATASETS_CACHE"] = "/tmp/.cache"
10
+
11
+ # Model mic, gratuit
12
  MODEL_NAME = "google/flan-t5-small"
13
 
14
  # Încarcă model și tokenizer
 
34
  inputs=gr.Textbox(lines=2, placeholder="Întreabă ceva..."),
35
  outputs="text",
36
  title="Chatbot simplu",
37
+ description="Chatbot minimal bazat pe Flan-T5-small (fără permisiuni la cache)."
38
  )
39
 
40
  iface.launch(server_name="0.0.0.0", server_port=7860)