Update app.py

app.py

@@ -11,6 +11,9 @@ trained_model.to(device)
 untrained_model.to(device)
 
 def generate(commentary_text, max_length, temperature):
+    if temperature <= 0:
+        return "Error: Temperature must be a strictly positive float.", "Error: Temperature must be a strictly positive float."
+    
     # Generate text using the finetuned model
     input_ids = trained_tokenizer(commentary_text, return_tensors="pt").input_ids.to(device)
     trained_output = trained_model.generate(input_ids, max_length=max_length, num_beams=5, do_sample=True, temperature=temperature)
@@ -29,7 +32,7 @@ iface = gr.Interface(
     inputs=[
         gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt"),
         gr.Slider(minimum=10, maximum=100, value=50, step=1,label="Max Length"),        
-        gr.Slider(minimum=0.01, maximum=2.0, value=0.7, label="Temperature")
+        gr.Slider(minimum=0.01, maximum=1.99, value=0.7, label="Temperature")
     ], 
     outputs=[
         gr.Textbox(label="commentary generation from finetuned GPT2 Model"),