wop commited on
Commit
db183cb
1 Parent(s): e8cf90a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -1,6 +1,14 @@
1
  import gradio as gr
2
  import os
3
- os.system('pip install transformers torch')
 
 
 
 
 
 
 
 
4
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
5
 
6
  # Load the pre-trained model and tokenizer
 
1
  import gradio as gr
2
  import os
3
+
4
+ # Check if transformers and torch are installed
5
+ try:
6
+ import transformers
7
+ import torch
8
+ except ImportError:
9
+ # Install transformers and torch if not installed
10
+ os.system('pip install transformers torch')
11
+
12
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
13
 
14
  # Load the pre-trained model and tokenizer