Update app.py

app.py

@@ -15,9 +15,9 @@ model_esm, alphabet = pretrained.load_model_and_alphabet('esm2_t36_3B_UR50D')
 model_esm.to('cuda')
 
 # Load the model
-model = Blip2ProteinMistral(config=FAPMConfig(), esm_size='3b')
-model.load_checkpoint("model/checkpoint_mf2.pth")
-model.to('cuda')
+# model = Blip2ProteinMistral(config=FAPMConfig(), esm_size='3b')
+# model.load_checkpoint("model/checkpoint_mf2.pth")
+# model.to('cuda')
 
 
 @spaces.GPU
@@ -94,9 +94,10 @@ def generate_caption(protein, prompt):
                'text_input': ['none'],
                'prompt': [prompt]}
     # Generate the output
-    prediction = model.generate(samples, length_penalty=0., num_beams=15, num_captions=10, temperature=1., repetition_penalty=1.0)
+    # prediction = model.generate(samples, length_penalty=0., num_beams=15, num_captions=10, temperature=1., repetition_penalty=1.0)
 
-    return prediction
+    # return prediction
+    return "test"
 
 # Define the FAPM interface
 description = """Quick demonstration of the FAPM model for protein function prediction. Upload an protein sequence to generate a function description. Modify the Prompt to provide the taxonomy information.