Update app.py

app.py

@@ -9,7 +9,7 @@ import requests
 from samplings import top_p_sampling, top_k_sampling, temperature_sampling
 from transformers import GPT2Config, GPT2Model, GPT2LMHeadModel, PreTrainedModel
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = torch.device("cpu")
 
 description = """
 <div>
@@ -23,7 +23,7 @@ description = """
 ## ℹ️ How to use this demo?
 1. Enter the control codes to set the musical form of the generated music. For details, please refer to the below "Control Codes" section. (optional)
 2. Enter the prompt of the generated music. You can set the control codes to set the musical form, ABC header (i.e., note length, tempo, meter, and key) and the motif of the melody. (optional)
-2. You can set the parameters (i.e., number of tunes, maximum patch, top-p, top-k, temperature and random seed) for the generation. (optional)
+2. You can set the parameters (i.e., number of tunes, maximum length, top-p, top-k, temperature and random seed) for the generation. (optional)
 3. Click "Submit" and wait for the result.
 4. The generated ABC notation can be played or edited using [EasyABC]([ABC Sheet Music Editor - EasyABC](https://easyabc.sourceforge.net/), you can also use this [Online ABC Player](https://abc.rectanglered.com/) to render the tune.
 
@@ -99,11 +99,6 @@ def generate_abc(prompt,
                  temperature,
                  seed):
 
-    if torch.cuda.is_available():    
-        device = torch.device("cuda")
-    else:
-        device = torch.device("cpu")
-
     tokenizer = ABCTokenizer()
     config = GPT2Config(vocab_size=len(tokenizer))
     model = GPT2LMHeadModel(config).to(device)