Update app.py
Browse files
app.py
CHANGED
@@ -2,8 +2,6 @@ import gradio as gr
|
|
2 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
3 |
import pytesseract
|
4 |
|
5 |
-
# Login to huggingface CLI
|
6 |
-
huggingface-cli login
|
7 |
|
8 |
# Initialize chat model (You can change the model here)
|
9 |
chat_model = pipeline("text-generation", model="gpt2") # You can switch to any model of your choice
|
|
|
2 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
|
3 |
import pytesseract
|
4 |
|
|
|
|
|
5 |
|
6 |
# Initialize chat model (You can change the model here)
|
7 |
chat_model = pipeline("text-generation", model="gpt2") # You can switch to any model of your choice
|