Update app.py

app.py

@@ -6,24 +6,33 @@ import os
 HF_TOKEN = os.getenv("HF_TOKEN")
 model_name = "Jonathanmann/GPT2-medium-SADnov21"
 
-# Load tokenizer and model from Hugging Face
-tokenizer = GPT2Tokenizer.from_pretrained(model_name, use_auth_token=HF_TOKEN)
-model = GPT2LMHeadModel.from_pretrained(model_name, use_auth_token=HF_TOKEN)
+# Load tokenizer and model from Hugging Face with error handling
+try:
+    tokenizer = GPT2Tokenizer.from_pretrained(model_name, use_auth_token=HF_TOKEN)
+    model = GPT2LMHeadModel.from_pretrained(model_name, use_auth_token=HF_TOKEN)
+except Exception as e:
+    raise RuntimeError(f"Error loading model or tokenizer: {e}")
 
-# Define the text generation pipeline
-generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
+# Define the text generation pipeline with error handling
+try:
+    generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
+except Exception as e:
+    raise RuntimeError(f"Error creating text generation pipeline: {e}")
 
-# Define a function for generating text
+# Define a function for generating text with error handling
 def generate_text(prompt, max_length, temperature, top_k, top_p):
-    response = generator(
-        prompt,
-        max_length=max_length,
-        temperature=temperature,
-        top_k=top_k,
-        top_p=top_p,
-        num_return_sequences=1
-    )
-    return response[0]["generated_text"]
+    try:
+        response = generator(
+            prompt,
+            max_length=max_length,
+            temperature=temperature,
+            top_k=top_k,
+            top_p=top_p,
+            num_return_sequences=1
+        )
+        return response[0]["generated_text"]
+    except Exception as e:
+        return f"An error occurred during text generation: {str(e)}"
 
 # Create the Gradio interface
 demo = gr.Interface(
@@ -40,5 +49,8 @@ demo = gr.Interface(
     description="A demo of Jonathanmann/GPT2-medium-SADnov21 with adjustable generation parameters."
 )
 
-# Launch the app
-demo.launch()
+# Launch the app with error handling
+try:
+    demo.launch()
+except Exception as e:
+    print(f"An error occurred while launching the Gradio interface: {str(e)}")