Update app.py

app.py

@@ -3,10 +3,12 @@ import gradio as gr
 import torch
 from transformers import pipeline
 
-pipe = transformers.pipeline("text-generation",
-                            model="marcelbinz/Llama-3.1-Minitaur-8B",
-                            device_map="auto",
-                            torch_dtype=torch.bfloat16)
+pipe = pipeline(
+    "text-generation",
+    model="marcelbinz/Llama-3.1-Minitaur-8B",
+    device_map="auto",
+    torch_dtype=torch.bfloat16
+)
 
 @spaces.GPU
 def infer(prompt):