Update app.py

app.py

@@ -1,3 +1,4 @@
+## app.py ##
 from transformers import pipeline
 from gradio import Interface
 import gradio as gr
@@ -14,7 +15,7 @@ MODELS = {
 def generate_and_analyze(model_name, input_text):
     # Load the model from the dictionary using the selected model name
     model = MODELS[model_name]
-    text_generator = pipeline('text-generation', model=model)
+    text_generator = pipeline('text-generation', model=model, device=0) # Use GPU if available
     result = text_generator(input_text, max_length=250, do_sample=True)[0]
     return result['generated_text']
 
@@ -28,3 +29,4 @@ iface = gr.Interface(
     outputs="text"
 )
 iface.launch()
+