Update app.py

app.py

@@ -14,7 +14,7 @@ MAX_LEN = number
 if st.button('Submit'):
 
     tokenizer = AutoTokenizer.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
-    model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator")
+    model = AutoModelForCausalLM.from_pretrained("ml6team/gpt-2-medium-conditional-quote-generator",use_cuda=False)
     inputs = tokenizer.encode(input_sequence, return_tensors='pt')
 
 # generate text until the output length (which includes the context length) reaches 50