Fix error in generate

app.py

@@ -20,12 +20,12 @@ import gradio as gr
 #iface.launch()
 
 def generate_model(name):
-    return f"Welcome to Gradio, {name}!"
+    return f"Welcome to Gradio HF_ACCES_TOKEN, {os.environ.get('HG_FACE_TOKEN')}!"
     
 def generate(prompt):
-    from diffusers import StableDiffusionPipeline
+    #from diffusers import StableDiffusionPipeline
     
-    pipe = StableDiffusionPipeline.from_pretrained("./output_model", torch_dtype=torch.float16)
+    #pipe = StableDiffusionPipeline.from_pretrained("./output_model", torch_dtype=torch.float16)
     pipe = pipe.to("cuda")
     image = pipe(prompt).images[0]  
     return(image)
@@ -36,7 +36,7 @@ def evaluate_model():
     #pipe = StableDiffusionPipeline.from_pretrained("./output_model", torch_dtype=torch.float16)
     #pipe = pipe.to("cuda")
     #image = pipe(prompt).images[0]  
-    return("Evaluate Model")
+    return(f"Evaluate Model {os.environ.get('HF_LLM_MODEL_ID')} from dataset {os.environ.get('HF_LLM_DATASET_ID')}")
 
 def train_model(*inputs):
     if "IS_SHARED_UI" in os.environ: